Closed sunstudio closed 4 years ago
Hello @sunstudio, 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 that 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:
For more information please visit https://www.ultralytics.com.
@sunstudio you can load a new dataset each epoch by placing the commands here within the epoch loop: https://github.com/ultralytics/yolov5/blob/01895f3924313424e6c7d5189e5c264cdd2f3ed2/train.py#L201-L203
thank you !
I want to use different trianing dataset in each epoch, these dataset are generated from a same dataset, imgaug can do the image generation , but I don't know where to call imgarg in train.py? Thanks a lot!