Skip to content

chris901003/DeepLearning

Repository files navigation

DeepLearning

pytorch_classification

圖像分類相關程式碼
目前實現的分類網路

  1. EfficientNet
  2. Mobilenet V2
  3. Mobilenet V3
  4. Resnet
  5. Swin-Transformer
  6. VGG
  7. VIT

pytorch_keypoint

人物關鍵點相關程式碼
目前實現的網路

  1. HRNet

pytorch_object_detection

目標檢測相關程式碼
同時還有實現DeepSort
目前實現網路

  1. Deformable-Detr
  2. Detr
  3. Faster-Rcnn
  4. Yolo V3
  5. Yolo V4
  6. Yolo X

pytorch_segmentation

語意分割相關程式碼
目前實現網路

  1. Deeplab V3
  2. FCN
  3. Lraspp

mmcv

官方Github
官方文檔
MMCV官方原始碼,目前是直接在官方原始碼上進行標注
MM系列的主要底層通用結構

mmsegmentation

官方Github
官方文檔
MMSegmentation官方原始碼,目前是直接在官方原始碼上進行標注
主要是處理分割項目

mmdetection

官方Github
官方文檔
MMDetection官方原始碼,目前是直接在官方原始碼上進行標注
主要是目標檢測

mmocr

官方Github
官方文檔
MMDetection官方原始碼,目前是直接在官方原始碼上進行標注
主要是文字方面處理

mmaction2

官方Github
官方文檔
MMAction2官方原始碼,目前是直接在官方原始碼上進行標注
主要是文字方面處理

mmpose

官方Github
官方文檔
MMPose官方原始碼,目前是直接在官方原始碼上進行標注
主要是文字方面處理

mmgeneration

官方Github
官方文檔
MMGeneration官方原始碼,為了可以在cpu上進行debug有改動原始碼,不確定是否可以進行正常訓練
不果可以進行debug,有需要可以自行重寫就行
主要是文字方面處理

learn_mmcv

學習mmcv官方原始碼,將需要的部分擷取出來,同時學習mmcv框架
暫時停止更新

pytorch_geometric

主要是在做圖相關的學習,目前是透過關節點的資訊辨識行為

pytorch_gan

主要是進行圖像生成相關學習

some_utils

有存放一些轉換用的程式工具

Test

一些測試文件