DeepLearning

介绍

个人学习项目,用于复现论文代码,深入理解算法原理。

注意:每个项目都可独立运行。若要运行某个项目,你需要将该项目作为根目录,以便找到对应模块。

classification

  • vision_transformer(完成)
  • mnist 手写字符(完成)
  • efficientNet(完成)
  • vggNet(完成)
  • resnet(完成)
  • coatNet(完成)
  • convNext(完成)
  • seNet(完成)

detection

  • RetinaNet(完成) 包含focal_loss
  • FPN(半完成) 实现resnet50 + fpn
  • YOLOV5 V5.0(完成) 实现注释,更新pt->onnx代码
  • yolox(完成) 修改了voc数据读取方式
  • FCOS (完成)

segmentation

  • FCN(完成)
  • U-Net(完成)

metric_learning

  • BDB(完成) 用于图像检索
  • Happy-Whale(完成) 鲸鱼竞赛检索baseline

self-supervised

  • MAE(完成) 实现VIT+MAE

deep_stereo

  • Real_time_self_adaptive_deep_stereo(实时双目里立体匹配,细节待完善)

other

  • tensorboard test(完成) 可视化网络,图片,训练过程以及卷积核
  • load weights test(完成) 权重部分加载
  • visual weights map test(完成) 特征图、卷积核可视化分析
  • label_convert(完成) 三种不同标注文件之间的转换以及box可视化
  • class_Activation_Map_Visual(完成) 可视化CNN的类激活图