Preparing dataset
Closed this issue · 1 comments
hfahrudin commented
Faced is really awesome and light! Can you tell me how is dataset shape and structure? i would like to train my yolo into just 5 classs
iitzco commented
Hi @hfahrudin !
Dataset was prepared using https://github.com/iitzco/widerface-to-tfrecord/blob/master/wider_to_tfrecord.py