Can I simple change the dataset to COCO?

hollance / YOLO-CoreML-MPSNNGraph

Tiny YOLO for iOS implemented using CoreML but also using the new MPS graph API.

MIT License

935 stars 252 forks source link

Can I simple change the dataset to COCO? #34

Closed ArchieGu closed 6 years ago

ArchieGu commented 6 years ago

I know the original Tiny-Yolo was trained on dataset VOC, however, I want to add more categories by using this net. Can I just simply change the dataset and retrain the model??