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

Reinforcement Learning Theory Book (rus)

ML-based medical imaging using Azure

Hierarchical User Intent Graph Network for Multimedia Recommendation

OpenDILab Multi-Agent Environment

Modifications of the official PyTorch implementation of StyleGAN3. Let's easily generate images and videos with StyleGAN2/2-ADA/3!

The codes and related files to reproduce the results for Image Similarity Challenge Track 1.

Repository for Traffic Accident Benchmark for Causality Recognition (ECCV 2020)

Code for visualizing the loss landscape of neural nets

PatchMatch-RL: Deep MVS with Pixelwise Depth, Normal, and Visibility

Fewshot-face-translation-GAN - Generative adversarial networks integrating modules from FUNIT and SPADE for face-swapping.

PyTorch/TorchScript compiler for NVIDIA GPUs using TensorRT

SAS: Self-Augmentation Strategy for Language Model Pre-training

Official implementation of the ICCV 2021 paper "Joint Inductive and Transductive Learning for Video Object Segmentation"

Implementation of the final project of the course DDA6309 Probabilistic Graphical Model

Intrusion Detection System using ensemble learning (machine learning)

CL-Gym: Full-Featured PyTorch Library for Continual Learning

Neon-erc20-example - Example of creating SPL token and wrapping it with ERC20 interface in Neon EVM

VISNOTATE: An Opensource tool for Gaze-based Annotation of WSI Data

This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.

Intrinsic Image Harmonization