pytorch image to vector (for Weaviate)

The inference container for the img2vec module

Documentation

Documentation for this module can be found here.

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

Swap linux/amd64 with linux/arm64 for arm-based builds

docker buildx build --platform linux/amd64 -t [url]/image-2-vector:[tag] -f Dockerfile.distroless --push .

Name		Name	Last commit message	Last commit date
Latest commit History 52 Commits
.github		.github
cicd		cicd
.dockerignore		.dockerignore
.gitignore		.gitignore
Dockerfile		Dockerfile
Dockerfile.distroless		Dockerfile.distroless
LICENSE		LICENSE
README.md		README.md
app.py		app.py
autoencoder.py		autoencoder.py
dimensionality_reduction.py		dimensionality_reduction.py
download_model.py		download_model.py
download_vit.py		download_vit.py
image2vec.py		image2vec.py
image2vec_vit.py		image2vec_vit.py
image_vector_dataset.py		image_vector_dataset.py
requirements-test.txt		requirements-test.txt
requirements.txt		requirements.txt
smoke_test.py		smoke_test.py
vectorizer.py		vectorizer.py