search

facebookresearch / unbiased-teacher

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection
https://arxiv.org/abs/2102.09480
MIT License
410 stars 82 forks source link

Augmentation? #30

Closed jiangzhengkai closed 2 years ago

jiangzhengkai commented 3 years ago

Hi, where is the augmentation implemeted? I could not find the weak aug and strong aug.

ycliu93 commented 2 years ago

This is where we implemented strong augmentation.

https://github.com/facebookresearch/unbiased-teacher/blob/6977c6f77c812fae4064dc1b3865658c2ed247b1/ubteacher/data/detection_utils.py#L9-L46

For the weak augmentation, we only use horizontal random flip. https://github.com/facebookresearch/unbiased-teacher/blob/6977c6f77c812fae4064dc1b3865658c2ed247b1/ubteacher/data/dataset_mapper.py#L92-L94

Thanks

ycliu93 commented 2 years ago

I will close this issue as there is no other question. Welcome to reopen if you have other relevant questions.