Skip to content

zaront/tensorflow-assetstore

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Making it easier to find assets

Demonstrating some possibilities using DNN and the asset store

Idea 1: categorize an asset by an image

What if you could have your asset automaticly tagged and categorized by an image?

Starting with a mobilenet model that was trained on imagenet I retrained the last layer of the convolutional neural net on images I web scraped from your asset store using a technique called transferred learning. This allowed me to leverage all the heavy lifting that went into recognizing real life objects and refine it to recognize the nuances that are present the 3D assets equivalents of those objects. The result is a very fast and light weight 3d asset recognizer, that is lightweight to scale and quick to retrain. It could easily achieve 400+ images per sec per instance or lightweight enough to auto-scale in the cloud

this demo is in the tagging folder

run python runtraining.py to train the model

run python score.py to visual inspect a collection of test images

  • test images can be placed in the ft_files/test folder

Alt text

for this demo I used only the first 2 pages from 7 diffrent asset categories, but the technique could be used to tag many more attributes.

Idea 2: find by similar image

What if you could search by an image and find assets that look visualy simmilar?

The work here is incomplete but so far I've extracted vectors from inception model for each asset image. (in the image_vectors folder) The next step would be to cluster them using nearest neighbor.

This demo is in the vector folder

This one is still work in progress but many thanks to the following as I piece this solution out:

inspired by this: https://blog.griddynamics.com/create-image-similarity-function-with-tensorflow-for-retail/

Idea 3: find by similar shape

What if assets you are using could be compared to assets in the store and similar "swap-out" assets that just came into the store could be recommended automaticly as possible improved replacements.

I havn't gotten to this idea yet but was inspired that converting the mesh to voxel and then running through a similar process as the "find by similar image" might work

inspired by this: https://people.csail.mit.edu/khosla/papers/cvpr2015_wu.pdf

About

image categorization using TensorFlow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages