Skip to content

Commit

Permalink
Update ReadMe
Browse files Browse the repository at this point in the history
more descriptions on inference, and more emoji
  • Loading branch information
Adamdad authored Mar 11, 2023
1 parent d75d76f commit 394b5f7
Showing 1 changed file with 20 additions and 11 deletions.
31 changes: 20 additions & 11 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ This repository contains the offical implementation for our CVPR-2023 paper.

**Consistent-Teacher: Towards Reducing Inconsistent Pseudo-targets in Semi-supervised Object Detection**

[[arxiv](https://arxiv.org/abs/2209.01589)] [[code](https://github.com/Adamdad/ConsistentTeacher)]
[[arxiv](https://arxiv.org/abs/2209.01589)] [[code](https://github.com/Adamdad/ConsistentTeacher)] [[project page](https://adamdad.github.io/consistentteacher/)]

Xinjiang Wang*, Xingyi Yang*, Shilong Zhang, Yijiang Li, Litong Feng, Shijie Fang, Chengqi Lyu, Kai Chen, Wayne Zhang

Expand All @@ -23,6 +23,7 @@ Xinjiang Wang*, Xingyi Yang*, Shilong Zhang, Yijiang Li, Litong Feng, Shijie Fan
![](assets/pipeline.jpg)

## Main Results
All results, logs, configs and checkpoints are listed here. Enjoy 👀!

**MS-COCO 1%/2%/5/%/10% Labeled Data**

Expand Down Expand Up @@ -100,6 +101,7 @@ Xinjiang Wang*, Xingyi Yang*, Shilong Zhang, Yijiang Li, Litong Feng, Shijie Fan
```
### Notes
- Defaultly, all models are trained on 8*V100 GPUs with 5 images per GPU.
- Additionally, we support the `2x8` and `fp16` training setting to ensure everyone is able to run the code, even with only 12G graphic cards.

## Usage

Expand Down Expand Up @@ -147,6 +149,19 @@ bash tools/dataset/prepare_coco_data.sh conduct

```
For concrete instructions of what should be downloaded, please refer to `tools/dataset/prepare_coco_data.sh` line [`11-24`](https://github.com/microsoft/SoftTeacher/blob/863d90a3aa98615be3d156e7d305a22c2a5075f5/tools/dataset/prepare_coco_data.sh#L11)

#### VOC0712 Dataset
- Download JSON files for unlabeled images PASCAL VOC data in COCO format
```
cd ${DATAROOT}
wget https://storage.cloud.google.com/gresearch/ssl_detection/STAC_JSON.tar
tar -xf STAC_JSON.tar.gz
# voc/VOCdevkit/VOC2007/instances_test.json
# voc/VOCdevkit/VOC2007/instances_trainval.json
# voc/VOCdevkit/VOC2012/instances_trainval.json
```

### Training

- To train model on the **partial labeled data** and **full labeled data** setting:
Expand All @@ -164,17 +179,11 @@ bash tools/dist_train.sh configs/consistent-teacher/consistent_teacher_r50_fpn_c

The core idea is to convert a new dataset to coco format. Details about it can be found in the [adding new dataset](https://github.com/open-mmlab/mmdetection/blob/master/docs/tutorials/customize_dataset.md).

#### VOC0712 Dataset
- Download JSON files for unlabeled images PASCAL VOC data in COCO format
```
cd ${DATAROOT}
### Inference and Demo
- To inference with the pretrained models on images and videos and plot the bounding boxes, we add two scripts
- `tools/inference.py` for image inference
- `tools/inference_vido.py` for video inference

wget https://storage.cloud.google.com/gresearch/ssl_detection/STAC_JSON.tar
tar -xf STAC_JSON.tar.gz
# voc/VOCdevkit/VOC2007/instances_test.json
# voc/VOCdevkit/VOC2007/instances_trainval.json
# voc/VOCdevkit/VOC2012/instances_trainval.json
```

## License

Expand Down

0 comments on commit 394b5f7

Please sign in to comment.