pybx

A simple python module to generate anchor (aka default/prior) boxes for object detection tasks.

retinopathy_classification_resnet18

Image classification for early detection of diabetic retinopathy in patients. This project uses a custom ResNet18 model built from scratch using PyTorch.

vit-pytorch

Implementation of ViT (vision transformer) paper using PyTorch

kaggle_melanoma

Melanoma classification using PyTorch and Lightning developed in nbdev