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.

Sign up for GitHub

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

Jump to bottom

How to do use test time augmentation? #95

Open

bravezzzzzz opened this issue Sep 6, 2021 · 1 comment

bravezzzzzz commented Sep 6, 2021

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?

Owner

PeizeSun commented Sep 8, 2021 •

edited

Loading

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment