How to train my own dataset?

rykov8 / ssd_keras

Port of Single Shot MultiBox Detector to Keras

MIT License

1.1k stars 553 forks source link

How to train my own dataset? #24

Closed zhanghan328 closed 7 years ago

zhanghan328 commented 7 years ago

I just want to detect the fish and capture it from a image. One picture only contain one kind of fish. My dataset has 7 kinds of fishes image. How to train my dataset and i can capture fish from a image. Thanks.

rykov8 commented 7 years ago

Please, see SSD_training.ipynb.

nixingyang commented 7 years ago

@zhanghan328 Hi, I suppose that you are working on the Kaggle competition. How is your progress with this SSD algorithm? From my experiment, the model suffered a lot from over-fitting and it could not detect fishes in the validation/testing dataset.