search

hhk7734 / tensorflow-yolov4

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

Can I use datasets in Yolo format for training? #59

Closed x12901 closed 3 years ago

x12901 commented 3 years ago

Great project. I use lableimg for labeling. But the project does not seem to support labels exported by this tool.Whether to support it in the future?

hhk7734 commented 3 years ago

labelImg supports yolo format dataset.

https://wiki.loliot.net/docs/lang/python/libraries/yolov4/python-yolov4-dataset#yolo

dataset = yolo.load_dataset("image_path.txt", image_path_prefix="/<prefix>", dataset_type="yolo")
hhk7734 commented 3 years ago

Did you use another format?

x12901 commented 3 years ago

Thank you.