search

shaimaaK / sticky-note-detector

SSD mobilebet v1 FPN pretrained model over COCO dataset, trained over sticky-note images dataset i built to detect sticky-notes
1 stars 1 forks source link

Sticky notes images #1

Open mchen07 opened 3 years ago

mchen07 commented 3 years ago

Would you mind posting your annotated sticky notes images? I am also planing to train some custom object detection model that can recognize sticky notes plus other objects. Your result is pretty good, if you can share your images, it would be really good.

shaimaaK commented 3 years ago

Done, I updated the repo with folder that contains all images I tested my model with and the code goes through all those images. Please let me know if you need anything else or have any questions.

mchen07 commented 3 years ago

Thanks for the immediate response. Sorry that I didn't describe clearly, since I am planning to train another custom object detector that can detect not only sticky notes, but also some other custom objects. I am trying to prepare some training data set, wondering if you can share your images for sticky notes that you have used to train your object detector, not the test images.

shaimaaK commented 3 years ago

Sure thing. I edited the readme file in Prepare The Dataset section, I included a link of all images used to train and validate the performance of the model before and after labeling.
TIP : the dataset I used is relatively small feel free to add more images, but make sure you have variety of sizes , orientation, colors to have more accurate detection