Category: Object Detection

This article explores the training of PyTorch RetinaNet model for traffic light detection in real-life scenes. ...

In this article, we train a custom PyTorch RetinaNet model and export it to ONNX format. Further, we run inference using it on the CUDA device. ...

In this article, we use PyTorch Faster RCNN and fine-tune it for small object detection on UAV images. ...

In this article, we use a PyTorch SSD model with custom ResNet34 backbone and train it on a person detection dataset. ...

In this article, we fine fine DETR (Detection Transformer) models on a custom aquarium dataset. After training, we also run inference on the test dataset and unseen videos. ...