Do you want your model to recognize only pens and pencils?
Just for an overview, you need to: * get labelled images of pens and pencils; * starting from these images, you have to [prepare a custom dataset](https://gluon-cv.mxnet.io/build/examples_datasets/detection_custom.html#sphx-glr-build-examples-datasets-detection-custom-py). I generally use [this program](https://github.com/tzutalin/labelImg) to manually select the bounding boxes locating the objects you want to detect in an image. This produces a .xml file for each image, then you can use them to generate the .lst and .rec files for the custom dataset. If you need, I have a program to generate the .rec files automatically, but you still have to manually label all the images, which is the boring part :) ; * once you have the dataset, you can [finetune a pre-trained model](https://gluon-cv.mxnet.io/build/examples_detection/finetune_detection.html). You can use a model pre-trained on the COCO dataset as a starting point, considering the objects should be similar; If you need any help, go ahead and ask. --- [Visit Topic](https://discuss.mxnet.io/t/retraining-ssd-for-pencils-and-pens/6528/2) or reply to this email to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.mxnet.io/email/unsubscribe/36a1aa1a6f2a2760083773fb717990055bc25b849ed6ce93ff9089d52ede5af9).
