/aerial-trash-detection-program

Full pipeline from downloading dataset to inferencing on test set

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Aerial Trash Detection

Full pipeline from downloading dataset to inferencing on test set in MMDetection. Kaggle is used for training the model but copies of the notebooks are provided here.

Usage

Requirements for Inference

Torch==1.13.0+cpu

Torchvision==0.14.0+cpu

MMDetection==2.22.0

MMCV==1.4.5

Opencv-headless (will not work with regular opencv-python)

Kaggle notebook links

https://www.kaggle.com/vexxingbanana/uavvaste-training

https://www.kaggle.com/vexxingbanana/uavvaste-dataset-creator-l

Credits

UAVVaste Dataset https://github.com/UAVVaste/UAVVaste