/UnoFlask

Primary LanguageJupyter Notebook

UnoFlask

Important!

First, you need to download the model itself in onnx extension. This can be done by clicking on the following link: Uno_detector.onnx. Then move the downloaded file to the "model" folder.


Running

  1. Download and run Docker.
  2. Execute the following commands:
  • Build Docker image using: docker build -t web-detector .
  • Run Docker container using: --rm --name detector -p 5000:5000 web-detector

Source code

  • model/Detect.py contains functions for processing the results obtained from the model.
  • tamplates contains the code responsible for the appearance of the home page.
  • Dockerfile contains instructions for starting the application.
  • Uno.ipynb contains code describing the learning process of the neural network.
  • main.py contains the logic of the site.

Example of interaction with the site.

  1. After starting Docker, go to http://localhost:5000. Image1
  2. Select the file and press the "Detect" button. Image2