Yolov5: How do I turn off color augmentation when traing?

[toilahung]

❔Question

I am training a YOLO for thermal images. I don't want to augment the color when training. What should I do to turn it off?

Thanks for your help.

Additional context

[github-actions[bot]]

Hello @toilahung, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook , Docker Image, and Google Cloud Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

• Cloud-based AI systems operating on hundreds of HD video streams in realtime.
• Edge AI integrated into custom iOS and Android apps for realtime 30 FPS video inference.
• Custom data training, hyperparameter evolution, and model exportation to any destination.

For more information please visit https://www.ultralytics.com.

[Aktcob]

https://github.com/ultralytics/yolov5/blob/master/utils/datasets.py#L563

[foodeveloperbar17]

Or you can change hyperparameters file,
https://github.com/ultralytics/yolov5/blob/7220cee1d1dc1f14003dbf8d633bbb76c547000c/data/hyp.finetune.yaml#L27
hsv_h: 0
hsv_s: 0
hsv_v: 0

[toilahung]

That's helpful. thank you!

[glenn-jocher]

Or you can change hyperparameters file,
https://github.com/ultralytics/yolov5/blob/7220cee1d1dc1f14003dbf8d633bbb76c547000c/data/hyp.finetune.yaml#L27

hsv_h: 0
hsv_s: 0
hsv_v: 0

This is the best-practices solution.

[xinxin342]

@topiaruss
How to get the output thermal diagram?
Can you teach me the specific operation?

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

[adhilcodes]

Turn off Hue, Saturation and exposure .
You can find it in the _YOLO cfg_ file