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

This repository contains the needed resources to build the HIRID-ICU-Benchmark dataset

Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch

🧑‍🔬 verify your TEAL program by experiment and observation

Catbird is an open source paraphrase generation toolkit based on PyTorch.

Sparse R-CNN: End-to-End Object Detection with Learnable Proposals, CVPR2021

Visualize Camera's Pose Using Extrinsic Parameter by Plotting Pyramid Model on 3D Space

Pretrained language model and its related optimization techniques developed by Huawei Noah's Ark Lab.

Look Closer: Bridging Egocentric and Third-Person Views with Transformers for Robotic Manipulation

Episodic Transformer (E.T.) is a novel attention-based architecture for vision-and-language navigation. E.T. is based on a multimodal transformer that encodes language inputs and the full episode history of visual observations and actions.

Gas detection for Raspberry Pi using ADS1x15 and MQ-2 sensors

This repository contains a CBIR system that uses swin transformer to extract image's feature.

RepVGG: Making VGG-style ConvNets Great Again

SOFT: Softmax-free Transformer with Linear Complexity, NeurIPS 2021 Spotlight

Zen-NAS: A Zero-Shot NAS for High-Performance Deep Image Recognition

Official repository for Few-shot Image Generation via Cross-domain Correspondence (CVPR '21)

Generating Digital Painting Lighting Effects via RGB-space Geometry (SIGGRAPH2020/TOG2020)

Unofficial pytorch implementation of 'Image Inpainting for Irregular Holes Using Partial Convolutions'

Satellite labelling tool for manual labelling of storm top features such as overshooting tops, above-anvil plumes, cold U/Vs, rings etc.

PyTorch Code for "Generalization in Dexterous Manipulation via Geometry-Aware Multi-Task Learning"

An Unsupervised Detection Framework for Chinese Jargons in the Darknet