search

hhk7734 / tensorflow-yolov4

YOLOv4 Implemented in Tensorflow 2.
MIT License
136 stars 75 forks source link

what about data augmentation? #46

Closed Adeel-Intizar closed 3 years ago

Adeel-Intizar commented 3 years ago

Does this implementation use data augmentation, i actually wanna make it more accurate, i wonder if data augmentation can be used in this implementation to generate much more accurate results?

hhk7734 commented 3 years ago

https://github.com/hhk7734/tensorflow-yolov4/blob/fd45eea11d2de86664978184e6b798daf2546cdc/py_src/yolov4/tf/__init__.py#L217-L237

https://github.com/hhk7734/tensorflow-yolov4/blob/fd45eea11d2de86664978184e6b798daf2546cdc/py_src/yolov4/tf/dataset.py#L311-L312

https://github.com/hhk7734/tensorflow-yolov4/blob/fd45eea11d2de86664978184e6b798daf2546cdc/py_src/yolov4/tf/dataset.py#L282-L294

If you want to use other augmentation, add to _next_random_augmentation_data().

Adeel-Intizar commented 3 years ago

Thank you! :)