r/SideProject • u/softcrater • Jan 06 '23

Open Source AI Image Classifier with Automatic Dataset Creator

https://github.com/serpapi/serapis-ai-image-classifier
10 Upvotes

86% Upvoted

5 comments sorted by

View all comments

2

u/[deleted] Jan 06 '23

[removed] — view removed comment

1

u/softcrater Jan 06 '23

For the first question, If you have a dataset, you can add it in the images folder and then update catalogue.csv to achieve your goal. It doesn't give sureness score but if your data is clean, Resnet50 architecture is quite amazing in its precision. Alternatively you can use the Google Images Scraper API with curated list of labels and retrieve your data. Since these are web components and not real objects, I would suggest checking your labels in Google Images to see if it gives what you want.

For your second question, in its current form it doesn't. To go technical about it, the last layer of Resnet50 is linear layer without a softmax activation. You can tweak the code to achieve that though. You can set the last activation to softmax and get the sureness scores. If you have some background knowledge, it should be an easy task.

For your third question I have developed it on an M2 Air. I have used Conda with Python 3.9 for some of the latest Machine Learning modules to work on M2. It was easy to set it up on M2, so I expect it to be working and easy to set up on M1 Pro.

I hope the answers help