- Prepare the data using script
python project/code/scripts/ClassifierDataSplitter.py --image_dir project/data/label_studio- Train the model using script
python project/code/notebooks/classifier_training.py- Copy the model to the models dir
cp -R classifier_model/image_classification_model/* project/code/yolo2class_pipeline/models/classificator- Run docker containers
cd project
docker compose up- Open the app in your browser
http://localhost:7860