This is a cnn based classification website with pytorch and flask, click the following image to watch the demo video.
run the following code to set up the image classification website.
$ python image_classification.py
This is an image classification website using flask and pytorch pre-trained model.
This is a cnn based classification website with pytorch and flask, click the following image to watch the demo video.
run the following code to set up the image classification website.
$ python image_classification.py
This is an image classification website using flask and pytorch pre-trained model.