Instant-Teaching: An End-to-End Semi-Supervised Object Detection Framework

Last update: Dec 31, 2022

Related tags

Overview

This repo is the official implementation of "Instant-Teaching: An End-to-End Semi-Supervised Object Detection Framework".

@inproceedings{zhou2021instant,
  title={Instant-Teaching: An End-to-End Semi-Supervised Object Detection Framework},
  author={Zhou, Qiang and Yu, Chaohui and Wang, Zhibin and Qian, Qi and Li, Hao},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={4081--4090},
  year={2021}
}

The code is based on MMdetection Toolbox.

1. Install

conda create -n instant_teaching python=3.7 -y
conda activate instant_teaching
conda install pytorch=1.7.1 torchvision cudatoolkit=10.1 -c pytorch
pip install mmcv==1.2.4

git clone http://gitlab.alibaba-inc.com/jianchong.zq/InstantTeaching.git
cd InstantTeaching
pip install -v -e .

2. Prepare COCO Dataset for Semi-Supervised Learning

mkdir -p datasets/coco
ln -s [coco_2017_downloaded] datasets/coco/coco_2017

python projects/InstantTeaching/tools/split_coco.py datasets/coco/coco_2017/annotations/instances_train2017.json 10

Two files of instances_train2017_10_labeled.json and instances_train2017_90_unlabeled.json will be generated.

3. Training

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch \
    --nproc_per_node=8 --master_port=1234 tools/train.py \
    projects/InstantTeaching/configs/instant_teaching_two_stage_r50_coco_quick_10.py \
    --launcher pytorch \
    --work-dir ${output_dir} \
    --no-validate

Instant-Teaching: An End-to-End Semi-Supervised Object Detection Framework

Related tags

Overview

1. Install

2. Prepare COCO Dataset for Semi-Supervised Learning

3. Training

Owner

YOLO5Face: Why Reinventing a Face Detector (https://arxiv.org/abs/2105.12931)

Codes and pretrained weights for winning submission of 2021 Brain Tumor Segmentation (BraTS) Challenge

Framework for Spectral Clustering on the Sparse Coefficients of Learned Dictionaries

Tutorials, assignments, and competitions for MIT Deep Learning related courses.

Code for our method RePRI for Few-Shot Segmentation. Paper at http://arxiv.org/abs/2012.06166

HiFT: Hierarchical Feature Transformer for Aerial Tracking (ICCV2021)

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two

Breast Cancer Classification Model is applied on a different dataset

Conservative Q Learning for Offline Reinforcement Reinforcement Learning in JAX

Production First and Production Ready End-to-End Speech Recognition Toolkit

Yet Another Robotics and Reinforcement (YARR) learning framework for PyTorch.

This repository implements variational graph auto encoder by Thomas Kipf.

Script utilizando OpenCV e modelo Machine Learning para detectar o uso de máscaras.

Multi-Output Gaussian Process Toolkit

A Python package for time series augmentation

Source code for GNN-LSPE (Graph Neural Networks with Learnable Structural and Positional Representations)

Official implementation of "Towards Good Practices for Efficiently Annotating Large-Scale Image Classification Datasets" (CVPR2021)

Framework for training options with different attention mechanism and using them to solve downstream tasks.

An implementation of Geoffrey Hinton's paper "How to represent part-whole hierarchies in a neural network" in Pytorch.

Implementation of paper "Self-supervised Learning on Graphs:Deep Insights and New Directions"