Object Detection using PyTorch for image using SSD | Single Shot Detection in Google Colab - Python

In this video, we are going to implement Object Detection in PyTorch for images. For PyTorch Object Detection, we will be using the SSD (Single Shot Detection )algorithm and the code will be available on Github. PyTorch framework has excelled in tasks like object detection in Computer Vision and carrying on its legacy, we have one of the most famous Object Detection algorithms SSD. The video basically focuses on detecting objects in an image and this can easily be scaled to multiple images and even videos. We use the pretrained version of SSD meaning that there is no fine tuning involved. We also use OpenCV for making bounding boxes for the detections. The whole project is implemented in Google Colab without any GPU. Image link: !wget 'images.cocodataset.org/val2017/000000037777.jpg' COCO Dataset names: coco_names = ["person" , "bicycle" , "car" , "motorcycle" , "airplane" , "bus" , "train" , "truck" , "boat" , "traffic light" , "fire hydrant" , "street sign" , "stop sign" , "parking meter" , "bench" , "bird" , "cat" , "dog" , "horse" , "sheep" , "cow" , "elephant" , "bear" , "zebra" , "giraffe" , "hat" , "backpack" , "umbrella" , "shoe" , "eye glasses" , "handbag" , "tie" , "suitcase" , "frisbee" , "skis" , "snowboard" , "sports ball" , "kite" , "baseball bat" , "baseball glove" , "skateboard" , "surfboard" , "tennis racket" , "bottle" , "plate" , "wine glass" , "cup" , "fork" , "knife" , "spoon" , "bowl" , "banana" , "apple" , "sandwich" , "orange" , "broccoli" , "carrot" , "hot dog" , "pizza" , "donut" , "cake" , "chair" , "couch" , "potted plant" , "bed" , "mirror" , "dining table" , "window" , "desk" , "toilet" , "door" , "tv" , "laptop" , "mouse" , "remote" , "keyboard" , "cell phone" , "microwave" , "oven" , "toaster" , "sink" , "refrigerator" , "blender" , "book" , "clock" , "vase" , "scissors" , "teddy bear" , "hair drier" , "toothbrush" , "hair brush"]