-
-
Notifications
You must be signed in to change notification settings - Fork 1.8k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Proof-of-concept for scene category classification #175
Comments
Moved our image classification code to the new Should be easier to test, simply go to the directory and run We'll see if that's a good name... was the best a could come up with today. |
Updated our docs: https://github.com/photoprism/photoprism/wiki/Image-Classification |
FYI: https://github.com/nic25 is working on this 🚀 |
Didn't find related models on TensorFlow Hub. |
seems like there's a way to convert places365 caffemodel to tensorflow? |
i successfully converted vgg16_hybrid1365 into pb file (took me half a day to get the converter working) and it works well in Tensorflow .NET. I'm now working on the non-hybrid model as it is more suitable for scene classification ResNet152-places365 (does not convert) |
The file is quite huge, ~500MB. Now shared on my nextcloud VGG16-places365. Its quite slow as my vps download speed is limited to 150KB. Label files are available on places365 github Some information on the model
Took a couple of images from the demo here and run it through the converted pb file in Tensorflow.NET. Image width & height set to 224px. Seems like conversion works really well |
Not sure if it's possible, but could the label "water" be added? Maybe "boat" for the second picture? I don't really know how those things work though 😕 |
As a PhotoPrism user, I want my photos classified by scene category so that I can filter search results by scene and get better image titles.
While we already label certain scenes based on the objects we find, we don't have a specialized TensorFlow model for this yet e.g. AlexNet-places365, GoogLeNet-places365 or ResNet152-places365. See also http://places2.csail.mit.edu/download.html.
Ideally we can reuse our existing Go TensorFlow code for this, but each model is different in how it must be used. Our NSFW detector for example needs different input values than our Nasnet model for object classification, so we ended up using different code and different packages. For scene detection, it might be good to create a new
scene
package unless merging it with Nasnet gives us much better performance.Acceptance Criteria:
The text was updated successfully, but these errors were encountered: