Skip to content
This repository has been archived by the owner on May 28, 2024. It is now read-only.

Update README.md #10

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 8 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@ consistency via strong augmentation.

<img src="docs/diagram.png" width="800">

__This code is only used for research. This is not an official Google product.__


# Instruction
Expand Down Expand Up @@ -65,17 +64,17 @@ wget http://models.tensorpack.com/FasterRCNN/ImageNet-R50-AlignPadding.npz

There are three steps:
- __1.__ Train a standard detector on labeled data
(`detection/scripts/coco/train_stg1.sh`).
[`detection/scripts/coco/train_stg1.sh`](./detection/scripts/coco/train_stg1.sh).
- __2.__ Predict pseudo boxes and labels of unlabeled
data using the trained detector (`detection/scripts/coco/eval_stg1.sh`).
data using the trained detector [`detection/scripts/coco/eval_stg1.sh`](./detection/scripts/coco/eval_stg1.sh).
- __3.__ Use labeled data and unlabeled data with
pseudo labels to train a STAC detector
(`detection/scripts/coco/train_stg2.sh`).
[`detection/scripts/coco/train_stg2.sh`](./detection/scripts/coco/train_stg2.sh).

Besides instruction at here, __`detection/scripts/coco/train_stac.sh`__
Besides instruction at here, [__`detection/scripts/coco/train_stac.sh`__](./detection/scripts/coco/train_stac.sh)
provides a combined script to train STAC.

__`detection/scripts/voc/train_stac.sh`__ is a combined script to train STAC on PASCAL VOC.
[__`detection/scripts/voc/train_stac.sh`__](./detection/scripts/voc/train_stac.sh) is a combined script to train STAC on PASCAL VOC.

The following example use labeled data as 10% train2017 and rest 90% train2017 data
as unlabeled data.
Expand Down Expand Up @@ -205,6 +204,9 @@ tensorboard --logdir=${PRJROOT}/detection/train_log
booktitle={arXiv:2005.04757}
}
```
## Disclaimer

This code is only used for research. This is not an official Google product.

# Acknowledgement

Expand Down