search

YangHai-1218 / RADet

Rigidity-Aware Detection for 6D Object Pose Estimation (CVPR 2023)
Apache License 2.0
41 stars 4 forks source link

Where to download COCO background images for the RandomBackground augmentation step? #8

Closed jfitzg7 closed 1 year ago

jfitzg7 commented 1 year ago

Hey @YangHai-1218,

I ran into an issue with the RandomBackground augmentation step specified in configs/base/datasets/bop_detection.py, which appears to be looking for background images in the data/coco directory by default. Just by looking at the code it seems like you could potentially use any set of images as the background. But just for the sake of it, do you guys know exactly which images you used and where to download them?

Also for the time being, I've just commented out the RandomBackground augmentation step in the configs/base/datasets/bop_detection.py file, would that be acceptable? Or is this a crucial augmentation step?

YangHai-1218 commented 1 year ago

Hey, sorry for the delay. You can comment this step if you don't have COCO dataset. In my oppion, it's better to use this step for better generalization. You can accesss these images by downloading COCO train 2017 dataset

jfitzg7 commented 1 year ago

No worries, thank you @YangHai-1218!