Download images:

python3 load_images.py

Train the model:

python3 train.py

Add a image (example: dog.jpg) to the root folder

Run the model:

python3 infer.py