Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Config files for evaluating the provided models #225

Open
Bai-YT opened this issue Aug 9, 2022 · 0 comments
Open

Config files for evaluating the provided models #225

Bai-YT opened this issue Aug 9, 2022 · 0 comments

Comments

@Bai-YT
Copy link

Bai-YT commented Aug 9, 2022

Hi. Is it possible to share the config files used for evaluating the weights available in through the Google Drive links?

I was trying to reproduce the 44.05% mAP of the Faster R-CNN (ResNet-50) -- Ours (thr=5e-2) experiment. However, I only get Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.324.

The command that I ran was the following:

bash tools/dist_test.sh
/home/ubuntu/project/Detection/SoftTeacher/configs/soft_teacher/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k_eval.py
/home/ubuntu/project/Detection/SoftTeacher/work_dirs/soft_teacher_faster_rcnn_r50_caffe_fpn_coco_full_720k/coco_iter_720000.pth
1 --eval bbox --cfg-options model.test_cfg.rcnn.score_thr=0.90

The config file is the following:

_base_="base.py"

data = dict(
    samples_per_gpu=8,
    workers_per_gpu=5,
    train=dict(
        sup=dict(
            ann_file="/home/ubuntu/project/data/COCO/annotations/instances_train2017.json",
            img_prefix="/home/ubuntu/project/data/COCO/train2017/",
        ),
    ),
    val=dict(
        ann_file="/home/ubuntu/project/data/COCO/annotations/instances_val2017.json",
        img_prefix="/home/ubuntu/project/data/COCO/val2017/",
    ),
    test=dict(
        ann_file="/home/ubuntu/project/data/COCO/annotations/instances_val2017.json",
        img_prefix="/home/ubuntu/project/data/COCO/val2017/",
    ),

    sampler=dict(
        train=dict(
            sample_ratio=[1, 1],
        )
    )
)

semi_wrapper = dict(
    train_cfg=dict(
        unsup_weight=2.0,
    )
)

optimizer = dict(lr=0.01, weight_decay=1e-4, momentum=0.9)
lr_config = dict(step=[300000, 425000])
runner = dict(_delete_=True, type="IterBasedRunner", max_iters=450000)

Could someone help me out? Thank you. If there is an existing issue about this that I missed, I apologize in advance.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant