Python影像辨識筆記(十六):使用PyTorch建立YOLOv3模型
3 min readJun 25, 2020
參考資料:
[ 0 ] ultralytics/yolov3
後續會基於該template去修改YOLOv3,因為是Github上使用PyTorch建立YOLOv3中,最多Star的。
[ 1 ] How to implement a YOLO (v3) object detector from scratch in PyTorch
從該系列的文章中發現,在PyTorch上建立YOLOv3是透過python去抓取yolov3.cfg神經網路架構資料成dict型態Part 1 : Understanding How YOLO works (主要是YOLOv3原理的介紹,沒有寫到程式碼)Part 2 (This one): Creating the layers of the network architecture (從yolov3.cfg檔案抓出網路架構資料成dict型態)Part 3 : Implementing the the forward pass of the network(建立Network)Part 4 : Objectness Confidence Thresholding and Non-maximum Suppression (處理閾值和NMS)Part 5 : Designing the input and the output pipelines(進行Image or Video的Detection)
[ 2 ] 匯總| Pytorch YOLO 項目推薦建議收藏學習
[ 3 ] [深度学习框架] PyTorch 常用代码段总结
[4] 利用PyTorch自己动手从零实现YOLOv3
[ 5 ] PyTorch 手册
[ 6 ] PYTORCH CHEAT SHEET
[ 7 ] deep-learning-v2-pytorch
推薦閱讀Introduction to Neural Networks部分
- Introduction to Neural Networks: Learn how to implement gradient descent and apply it to predicting patterns in student admissions data.
- Sentiment Analysis with NumPy: Andrew Trask leads you through building a sentiment analysis model, predicting if some text is positive or negative.
- Introduction to PyTorch: Learn how to build neural networks in PyTorch and use pre-trained networks for state-of-the-art image classifiers.
[ 8 ] YOLO-Reproduce-Summary
蒐集了YOLO透過各種不同Framework的實現
[ 9 ] PyTorch Computer Vision Cookbook
書中的Multi-Object Detection介紹了如何使用PyTorch實現YOLOv3
[ 10 ]【从零开始学习YOLOv3】7. 教你在YOLOv3模型中添加Attention机制
介紹如何在現有的PyTorch架構下,加入其他的Tricks