/image_classification

Framework for training and tracking deep learning models built on pytorch-lightning and hydra. Soon to encompass both image classification, semantic segmentation and object detection.

Primary LanguagePython

Stargazers