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

Adaout is a practical and flexible regularization method with high generalization and interpretability

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2020

A community run, 5-day PyTorch Deep Learning Bootcamp

codebase for "A Theory of the Inductive Bias and Generalization of Kernel Regression and Wide Neural Networks"

Implementation of Research Paper "Learning to Enhance Low-Light Image via Zero-Reference Deep Curve Estimation"

A lightweight deep network for fast and accurate optical flow estimation.

CLIPort: What and Where Pathways for Robotic Manipulation

Semantic Segmentation Architectures Implemented in PyTorch

Le dataset des images du projet d'IA de 2021

Safe Control for Black-box Dynamical Systems via Neural Barrier Certificates

A DNN inference latency prediction toolkit for accurately modeling and predicting the latency on diverse edge devices.

ComputerVision - This repository aims at realized easy network architecture

Tiny-NewsRec: Efﬁcient and Effective PLM-based News Recommendation

Rl-quickstart - Reinforcement Learning Quickstart

A PyTorch implementation of SIN: Superpixel Interpolation Network

Functional TensorFlow Implementation of Singular Value Decomposition for paper Fast Graph Learning

Code of paper: "DropAttack: A Masked Weight Adversarial Training Method to Improve Generalization of Neural Networks"

An implementation of the proximal policy optimization algorithm

Deep Learning with PyTorch made easy 🚀 !

Graph Transformer Architecture. Source code for