Machine-Learning-Models-Review

A implementation of Deep Learning models using the deep learning framwork Pytorch.

Image Classification Models

Models implemented on /image_classification/models.

• AlexNet. https://papers.nips.cc/paper_files/paper/2012
• GoogleNet.https://doi.org/10.48550/arXiv.1409.4842
• InceptionResNet. https://doi.org/10.48550/arXiv.1602.07261
• NiN. https://doi.org/10.48550/arXiv.1312.4400
• VggNet. https://doi.org/10.48550/arXiv.1409.1556

Datasets:

Dataset employed for comparison:

• MiniImageNet: https://cseweb.ucsd.edu/~weijian/static/datasets/mini-ImageNet

Semantic Segmentation Models

Models implemented on /semantic_segmentation/models.

• FullyConvNet. https://arxiv.org/abs/1511.00561
• SegNet. https://arxiv.org/abs/1411.4038

Dataset

Dataset employed for comparison:

• FreiHand: https://lmb.informatik.uni-freiburg.de/resources/datasets/FreihandDataset.en.html