search

PeizeSun / SparseR-CNN

[CVPR2021, PAMI2023] End-to-End Object Detection with Learnable Proposal
MIT License
1.32k stars 187 forks source link

How to do use test time augmentation? #95

Open bravezzzzzz opened 3 years ago

bravezzzzzz commented 3 years ago

hello, I want to run inference with test-time augmentation. I set TEST.AUG.ENABLED to true, but it is not working. Can you share how to deal with it?

PeizeSun commented 3 years ago

Hi~ The default TTA code in detectron2 doesn't work for sparse r-cnn. We have uploaded the right TTA code. You can use it by

python projects/SparseRCNN/train_net.py --num-gpus 8 --config-file projects/SparseRCNN/configs/sparsercnn.res50.100pro.3x.yaml --eval-only MODEL.WEIGHTS path/to/model.pth TEST.AUG.ENABLED True