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

Dynamic Attentive Graph Learning for Image Restoration, ICCV2021 [PyTorch Code]

This implements the learning and inference/proposal algorithm described in "Learning to Propose Objects, Krähenbühl and Koltun"

The official repo of the CVPR2021 oral paper: Representative Batch Normalization with Feature Calibration

Complete-IoU (CIoU) Loss and Cluster-NMS for Object Detection and Instance Segmentation (YOLACT)

Experiments and code to generate the GINC small-scale in-context learning dataset from "An Explanation for In-context Learning as Implicit Bayesian Inference"

Spam your friends and famly and when you do your famly will disown you and you will have no friends.

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder

A repo for Causal Imitation Learning under Temporally Correlated Noise

ViDT: An Efficient and Effective Fully Transformer-based Object Detector

Numenta published papers code and data

TF2 implementation of knowledge distillation using the "function matching" hypothesis from the paper Knowledge distillation: A good teacher is patient and consistent by Beyer et al.

LV-BERT: Exploiting Layer Variety for BERT (Findings of ACL 2021)

Code repository accompanying the paper "On Adversarial Robustness: A Neural Architecture Search perspective"

A very lightweight monitoring system for Raspberry Pi clusters running Kubernetes.

Code for ACL 2019 Paper: "COMET: Commonsense Transformers for Automatic Knowledge Graph Construction"

SMD-Nets: Stereo Mixture Density Networks

Pytorch Lightning Distributed Accelerators using Ray

A Dying Light 2 (DL2) PAKFile Utility for Modders and Mod Makers.

Pytorch implementation of PCT: Point Cloud Transformer

Simple torch.nn.module implementation of Alias-Free-GAN style filter and resample