How to turn off albumentations?

THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]

https://arxiv.org/abs/2405.14458

GNU Affero General Public License v3.0

9.58k stars 913 forks source link

Closed Andy718811 closed 3 months ago

Andy718811 commented 3 months ago

How to turn off the albumentations methods that always used in yolov10?

Blur(p=0.01, blur_limit=(3, 7)), MedianBlur(p=0.01, blur_limit=(3, 7)), ToGray(p=0.01), CLAHE(p=0.01, clip_limit=(1, 4.0), tile_grid_size=(8, 8))

leonnil commented 3 months ago

Thank you for your interest! I think you can just run pip uninstall albumentations, or try commenting out the code here: https://github.com/THU-MIG/yolov10/blob/2c36ab0f108efdd17c7e290564bb845ccb6844d8/ultralytics/data/augment.py#L1002

Andy718811 commented 3 months ago

Thank you so much for your reply, it worked well.