Novel Instances Mining with Pseudo-Margin Evaluation for Few-Shot Object Detection

Last update: Sep 08, 2022

Overview

Novel Instances Mining with Pseudo-Margin Evaluation for Few-Shot Object Detection (NimPme)

The official implementation of Novel Instances Mining with Pseudo-Margin Evaluation for Few-Shot Object Detection

The code is built on TFA

Requirements

Linux with Python >= 3.6
PyTorch >= 1.4
torchvision that matches the PyTorch installation
CUDA 10.0, 10.1, 10.2
GCC >= 4.9

Getting Started

Evaluation with pre-trainied 10-shot final detecor

we provide the pre-trainied 10-shot final detecor

python3 -m tools.test_net --num-gpus 1 \
        --config-file configs/PascalVOC-detection/split1/faster_rcnn_R_101_FPN_ft_all1_10shot.yaml \
        --eval-only

Training & Evaluation in Command Line

To train a base detector, run

python3 -m tools.train_net --num-gpus 1 \
        --config-file configs/PascalVOC-detection/split1/faster_rcnn_R_101_FPN_base1.yaml

fine-tune the detector with novel set

python3 -m tools.ckpt_surgery \
        --src1 checkpoints/coco/faster_rcnn/faster_rcnn_R_101_FPN_base/model_final.pth \
        --method randinit \
        --save-dir checkpoints/coco/faster_rcnn/faster_rcnn_R_101_FPN_all
        --coco

python3 -m tools.train_net --num-gpus 1 \
        --config-file configs/COCO-detection/faster_rcnn_R_101_FPN_ft_all_1shot.yaml \
        --opts MODEL.WEIGHTS $WEIGHTS_PATH

fine-tune the detector with pseudo set

python3 -m tools.genarate_pseudo --num-gpus 1
python3 -m tools.train_feature --num-gpus 1

Novel Instances Mining with Pseudo-Margin Evaluation for Few-Shot Object Detection

Related tags

Overview

Novel Instances Mining with Pseudo-Margin Evaluation for Few-Shot Object Detection (NimPme)

Getting Started

Evaluation with pre-trainied 10-shot final detecor

Training & Evaluation in Command Line

Owner

MRQy is a quality assurance and checking tool for quantitative assessment of magnetic resonance imaging (MRI) data.

Using VapourSynth with super resolution models and speeding them up with TensorRT.

Semi-supervised Stance Detection of Tweets Via Distant Network Supervision

Share a benchmark that can easily apply reinforcement learning in Job-shop-scheduling

A Python library for Deep Probabilistic Modeling

pytorch implementation of fast-neural-style

[AAAI 2022] Negative Sample Matters: A Renaissance of Metric Learning for Temporal Grounding

Practical and Real-world applications of ML based on the homework of Hung-yi Lee Machine Learning Course 2021

A new GCN model for Point Cloud Analyse

Header-only library for using Keras models in C++.

Web service for facial landmark detection, head pose estimation, facial action unit recognition, and eye-gaze estimation based on OpenFace 2.0

Forecasting with Gradient Boosted Time Series Decomposition

A high-performance distributed deep learning system targeting large-scale and automated distributed training.

A Comparative Review of Recent Kinect-Based Action Recognition Algorithms (TIP2020, Matlab codes)

Data and Code for ACL 2021 Paper "Inter-GPS: Interpretable Geometry Problem Solving with Formal Language and Symbolic Reasoning"

EdiBERT is a generative model based on a bi-directional transformer, suited for image manipulation

Reliable probability face embeddings

This repository includes the code of the sequence-to-sequence model for discontinuous constituent parsing described in paper Discontinuous Grammar as a Foreign Language.

Massively parallel Monte Carlo diffusion MR simulator written in Python.

VITS: Conditional Variational Autoencoder with Adversarial Learning for End-to-End Text-to-Speech