FasterRCNN代碼解讀1

之前的文章簡要介紹了Faster-RCNN等物體檢測的算法,本文將從代碼角度詳細分析介紹Faster-RCNN的實現。本文使用的代碼參考了chenyuntc的實現,代碼的位置看這裏。需要注意的是,本文使用的框架是Pytorch。 數據載入 數據載入部分的代碼主要見./data/dataset.py中的類Dataset與TestDataset。 數據載入部分的邏輯如下: 從VOC數據集中獲得img,
相關文章
相關標籤/搜索