mirrors
/
RE-OWOD
mirror of https://github.com/RE-OWOD/RE-OWOD
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update README.md

main
RE-OWOD 2022-06-19 20:03:06 +08:00 committed by GitHub
parent c9cba21525
commit d0a04a5b90
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
README.md
View File

@ -12,6 +12,48 @@ For the testing set, we adopt the validation set of COCO2017, which contains rel
The data files are at [./datasets/Main/](./datasets/Main/).
<!-- And the codes for creating these files are at [datasets/coco_utils](datasets/coco_utils) -->
## Train
Train with PAD:
The PAD module is implemented in [./detectron2/modeling/roi_heads/roi_heads.py](./detectron2/modeling/roi_heads/roi_heads.py),where the file generated by selective search needs to be loaded. The code that generates these files is in the [./datasets/save_selective_search.py](./datasets/save_selective_search.py).
## Inference
Inference with CEC
```
python tools/train_net.py --num-gpus 3\
--eval-only --config-file ./configs/OWOD/t1/t1_test.yaml \
SOLVER.IMS_PER_BATCH 6 SOLVER.BASE_LR 0.005 \
OUTPUT_DIR "./output/t1/test_dir/" \
MODEL.WEIGHTS "/Path/To/Your/Model" \
OWOD.GENERATE_CALI False \
OWOD.CALI_PATH "/home/RE-OWOD-main/analysis/t1_REOWOD_train_scores_cali_07" \
OWOD.UNK_THRESH 0.6
```
Inference without CEC
```
python tools/train_net.py --num-gpus 3\
--eval-only --config-file ./configs/OWOD/t1/t1_test.yaml \
SOLVER.IMS_PER_BATCH 6 SOLVER.BASE_LR 0.005 \
OUTPUT_DIR "./output/t1/test_dir/" \
MODEL.WEIGHTS "/Path/To/Your/Model" \
OWOD.GENERATE_CALI True
```
If you want to generate CEC pickle file with train datasets, you need to repalce "./detectron2/evaluation/evaluator.py" with "./detectron2/evaluation/evaluator_cali.py".
Remember to add current stage to CEC pickle files' name in "./detectron2/evaluation/evaluator_cali.py".
```
python tools/train_net.py --num-gpus 3\
--eval-only --config-file ./configs/OWOD/t1/t1_test.yaml \
SOLVER.IMS_PER_BATCH 6 SOLVER.BASE_LR 0.005 \
OUTPUT_DIR "./output/t1/test_dir/" \
MODEL.WEIGHTS "/Path/To/Your/Model" \
OWOD.GENERATE_CALI True
```
<!--
## Citation