It is required to create the data and output folder on the device before starting work, and place the images there for detective work. By default, 3 standard ones are downloaded, on which you can test the system.
Run project
docker-compose up
Instructions for each of the modules in the corresponding directory
Requirements for use:
- Python >= 3.6
If you need to use the docker image of this service directly, then use
-
Cpu version
docker run -ti sapfir0/premier-eye
-
Gpu version
docker run -ti sapfir0/premier-eye:gpu
Main command:
python3 mainImage.py
Of course, on windows the command will look like
python mainImage.py
Run algorithms tests
python3 -m unittest tests/detections.py
If you want run container on interactive mode
docker run -it sapfir0/premier-eye:cpu bash^C
- Algorithm 1. Mask R-CNN
- Implementation in neural_network/mask
- Example of work
- Algorithm 2. ImageAI deprecated
- Implementation in neural_network/imageAI
- Example of work