kangchengX/CNN-ViT

Implements MobileViT and ViT from scratch, compatible with graph execution mode in tensorflow. Compares multiple cnn and transformer-based models including VGG, ResNet, ViT and MobileViT on a small data set.

Python