sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
223 stars 77 forks source link

How to disable Image Augmentations in Faster RCNN #38

Closed samahwaleed closed 1 year ago

samahwaleed commented 1 year ago

I want to disable the image augmentation, could you please show me how!

Thank you!

sovit-123 commented 1 year ago

@samahwaleed The color and blur augmentations are disabled by default. You need to pass --use-train-aug to the training script to enable it. Mosaic augmentations are always applied by default. In case you don't want to use any augmentation, you need to pass the command similar to the following:

python train.py --no-mosaic <rest of the command>

samahwaleed commented 1 year ago

Thank you Sir,

sovit-123 commented 1 year ago

Welcome