Migrate knn-image-classifier from tfjs-core to tfjs-models

tensorflow / tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.

https://js.tensorflow.org

Apache License 2.0

18.49k stars 1.93k forks source link

Migrate knn-image-classifier from tfjs-core to tfjs-models #385

Closed dsmilkov closed 6 years ago

dsmilkov commented 6 years ago

The KNN classifier lives in tfjs-core and is outdated.

We need to make it generic (have it work with any embedding, not just squeezenet) and move it to tfjs-models. Then show a demo that uses MobileNet embeddings + KNN.

HalfdanJ commented 6 years ago

Has been added to tfjs-models now, so can be removed from tfjs-core