FurkanOM / tf-faster-rcnn

Tensorflow 2 Faster-RCNN implementation from scratch supporting to the batch processing with MobileNetV2 and VGG16 backbones
Apache License 2.0
94 stars 61 forks source link

HOW to train my own dataset? #10

Open MC1016 opened 3 years ago

MC1016 commented 3 years ago

how i can use my own dataset like voc2007 which include train.txt test.txt?

FurkanOM commented 3 years ago

You can use custom datasets by creating your own data generator. You can find examples of them in tensorflow tutorials. You can also use the custom_data_generator I used for prediction as a starting point. https://github.com/FurkanOM/tf-faster-rcnn/blob/db54f3d873d74cec50b9d21409dcb831f271b7bb/utils/data_utils.py#L120