search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.24k stars 1.24k forks source link

does the model support mosaic data enhancement #321

Open joe660 opened 3 years ago

joe660 commented 3 years ago

Hello, does the model support mosaic data enhancement？

cuteboyqq commented 2 years ago

you can use YoloV4 or YoloV5 mosaic class (getitem function will output all mosaic images and corresponding labels) to generate mosaic data, but you need to translate labels to label.txt here is my source code https://github.com/cuteboyqq/image_processing/blob/main/mosaic.py