Test

Now that you’ve got your data in a format the model expects, let’s train an EfficientDet to do object detection on our dataset.

Within the root directory of the project, run the following command

python trashnet/test.py --path DATA_PATH --pretrained PRETRAINED_PATH --output OUTPUT_PATH

where DATA_PATH is the path to the COCO style dataset that you set up previously, PRETRAINED_PATH is the path to the pretrained models from the training loop and OUTPUT_PATH is the path you want to save the predicted images to.

For additional hyperparameter choices available during training, use help

python trashnet/test.py --help

When altering the hyperparametrs for the testing loop, to get the best results, try to be consistent with the hyperparameter choices for the training loop.