├── Artifacts └──best.onnx ├── predict_data ├── predicted_images ├── Utility_dir └──models └──utils ├── README.md └── src ├── predict_pipeline.py ├── get_description.py ├── create_csv.py └── Config.yaml
python -m venv env
pip install -r requirements.txt
We made the prediction pipeline to run both single image and for image directory
python predict_pipeline.py --input_path path --output_dir path
python predict_pipeline.py --input_path predict_data --output_dir predicted_images
python predict_pipeline.py --input_path predict_data/DJI_0185.jpg --output_dir predicted_images