YOLO個人解讀和部分代碼註釋

YOLOv1 YOLOv2 個人解讀和部分代碼註釋

​ 前一段時間過年那陣子，做了一個小項目，用YOLO檢測車輛。於是把yolo的論文v1 v2刷了一遍，然後下了代碼，跑了一遍。後期老師說讓我好好去讀讀源代碼，我花了幾天時間把基於darknet的代碼刷了一遍，感受頗多。覺得大神不僅學術棒，代碼寫的也是非常的好。本來打算早早把這些記錄下來，無奈事情多，還上課，一拖再拖。

​ 本次重點把自己註釋的代碼拉出來，希望大家一起學習，肯定有很多錯誤，希望大神們指正。

YOLO將之前常用的分類問題改爲迴歸問題，將原來的two-stage變成one-stage。

重點註釋了

• convolutional_layer.c 卷積層 比較難理解

先上圖，可以看一下這個鏈接

• detection_layer.c 這個是yolo_v1裏面最核心的，計算損失函數

• dropout_layer.c

• gemm.c 矩陣相乘

• im2col.c

• maxpool_layer.c

• region_layer.c 這個是yolo_v2裏面最核心的，計算損失函數。

註釋的代碼在我的github