/PyTorchImageDetector

A Python script on PyTorch that detects any objects present in an image.

Primary LanguagePython

This is a PyTorch model for detecting recognizable objects in an image. I used the COCO dataset's labels for the object labels for the model.

Run the following command on terminal to obtain the labels for any objects detected in the image.

python detect.py --images image.png --det det