Computer Vision Model Implementation by Pytorch

This is the repo including multiple Computer Vision Model Implementation with full explanation. For each model, there is tutorial for any operation.

Model with task category:

Image Classification

1. AlexNet
2. ResNet
3. VGG
4. ViT

2D Semantic Segmentation

1. Unet++
2. DeepLabV3
3. FCN

3D Semantic Segmentation

1. PointNet

2D Object Detection

1. Faster-RCNN (In progress)

6D Pose Estimation

1. PoseCNN