minor fix in readme

pull/3/head
Dahun Kim 2021-08-31 16:52:13 +09:00
parent 18cf7c03b1
commit e004d26165
2 changed files with 5 additions and 4 deletions

View File

@ -14,7 +14,7 @@ Humans can recognize novel objects in this image despite having never seen them
## Performance on COCO-Split setting
### VOC to non-VOC cross-category generalization
### Cross-category generalization
We train OLN on COCO VOC categories, and test on non-VOC categories. Note our AR@k evaluation does not count those proposals on the 'seen' classes into the budget (k), to avoid evaluating recall on see-class objects.
@ -67,7 +67,8 @@ Our trained models are available for download [here](https://drive.google.com/uc
```
# Multi-GPU distributed testing
bash tools/dist_test_bbox.sh configs/oln_box/oln_box.py trained_weights/latest.pth ${NUM_GPUS}
bash tools/dist_test_bbox.sh configs/oln_box/oln_box.py \
trained_weights/latest.pth ${NUM_GPUS}
```

View File

@ -75,7 +75,7 @@ model = dict(
nms_pre=2000,
nms_post=2000,
max_num=2000,
nms_thr=0.7,
nms_thr=1.0,
min_bbox_size=0),
rcnn=dict(
assigner=dict(
@ -103,7 +103,7 @@ model = dict(
min_bbox_size=0),
rcnn=dict(
score_thr=0.0,
nms=dict(type='nms', iou_threshold=0.5),
nms=dict(type='nms', iou_threshold=0.7),
max_per_img=1500,
)))
dataset_type = 'CocoSplitDataset'