pytorch image to vector (for Weaviate)

The inference container for the img2vec module

Documentation

Documentation for this module can be found here.

Build Docker container

LOCAL_REPO="img2vec-pytorch" MODEL_NAME="resnet50" ./cicd/build.sh