YueLiao / CDN

Code for "Mining the Benefits of Two-stage and One-stage HOI Detection"
Apache License 2.0
89 stars 15 forks source link

The model providing is not the best. #15

Closed Devotedyang closed 2 years ago

Devotedyang commented 2 years ago

Hi~ the CDN-L(R101) model you provide show the result of 63.01(Scenario 1) and 64.45(Scenario 2), which is not same as the result 63.91 and 65.89 in paper. So could you provide the model with best performance? Thanks a lot.

YueLiao commented 2 years ago

Could you provide your test/inference script? I think some wrong args setting makes the accuracy misalignment.

Devotedyang commented 2 years ago

Could you provide your test/inference script? I think some wrong args setting makes the accuracy misalignment.

I just download the model and code with no changes, and then use the following commands: python -m torch.distributed.launch \ --nproc_per_node=4 \ --use_env \ main.py \ --pretrained pretrained/vcoco_cdn_l.pth \ --batch_size 8 \ --dataset_file vcoco \ --hoi_path data/v-coco \ --num_obj_classes 81 \ --num_verb_classes 29 \ --backbone resnet101 \ --num_queries 100 \ --dec_layers_hopd 6 \ --dec_layers_interaction 6 \ --eval \ --use_nms_filter With the commands above, it shows 63.01(Scenario 1) and 64.45(Scenario 2).

YueLiao commented 2 years ago

This script only provides a reference result, like mAP all: 0.6332 mAP thesis: 0.6480, which is not the final performance evaluated by the official script. You need to follow the official evaluation to obtain s1 and s2 AP. Otherwise, the performance is affected a little by the different batch sizes.

Devotedyang commented 2 years ago

This script only provides a reference result, like mAP all: 0.6332 mAP thesis: 0.6480, which is not the final performance evaluated by the official script. You need to follow the official evaluation to obtain s1 and s2 AP. Otherwise, the performance is affected a little by the different batch sizes.

Thanks! I'm new to COCO/V-COCO, your answer is very helpful.

YueLiao commented 2 years ago

If any issues, feel free to re-open.