/DeepMushroom-docker

Docker container of DeepMushroom

Primary LanguagePython

DeepMushroom-docker

DeepMushroom-docker is a docker container running the deep learning mushroom classifier from DeepMushroom project.

Prepare && Getting Started

Save the trained PyTorch and place it inside the directory \model\model.pth

Build and run the docker container:

$ docker-compose build & docker-compose up