SSD | PyTorch
pytorch.org › hub › nvidia_deeplearningexamples_ssdRun the SSD network to perform object detection. with torch.no_grad(): detections_batch = ssd_model(tensor) By default, raw output from SSD network per input image contains 8732 boxes with localization and class probability distribution. Let’s filter this output to only get reasonable detections (confidence>40%) in a more comprehensive format.