Category: PyTorch

In this article, we explore the DETR model for object detection. Along with a short discussion of the model architecture, we also carry out inference on videos. ...

In this article, we train the PyTorch RetinaNet model on a custom BCCD dataset. We go through the configuration of the RetinaNet model for custom training, train the model, and also carry out inference. ...

In this article, we go through the training of SSD300 VGG16 from Torchvision on a custom license plate detection dataset. ...

In this article, we train several Faster RCNN object detection models for to detect defects on the steel surface. ...

In this article, we train a simple embedding language model on disaster tweet classification dataset and then run inference on the test set. ...