基於pytorch的SSD復現筆記

2021-09-12 02:06:19 字數 942 閱讀 7681

復現的過程按照github的步驟進行:

但注意有幾個坑:

sudo apt install curl
兩個命令的使用參考:

2.cv2的安裝:

詳細參考我的另一篇部落格:

cv2的安裝包名字不叫cv2,而是叫opencv-python

3.pytorch版本需要為0.4.1,python版本為python3.6(千萬別用3.70)

python3.6的anaconda安裝,見我的一篇部落格:

4.loss計算,需要加一句:

在multibox_loss.py的97行前加:

loss_c = loss_c.view(num, -1)#the line added

loss_c[pos] = 0 # filter out pos boxes for now

5.加上之後

n會報資料型別有問題,

n = num_pos.data.sum().float()
做完這幾步就可以訓練了。

這個程式中eval和test都沒有將檢測結果框視覺化,但在demo資料夾中提供了demo的**,在使用這部分**的時候有乙個bug,總是報記憶體不足。

解決辦法:將python3.6切換到python2.7

結果:

從結果上可以看到對於多人檢測還是有兩個很明顯的人沒有檢測出來。

pytorch目標檢測ssd二 ssd的整體框架

這是ssd的第二篇部落格,主要是介紹ssd的整體框架,不會涉及 這篇部落格,我會分析一下整體框架。1 輸入 輸入的大小要被設定成300x300的,所以不是300x300的,輸入之前需要resize一下 2 特徵提取網路 深度神經網路可以進行特徵提取,我們的ssd採用的就是vgg深度神經網路。但是ss...

經典GANs模型的pytorch復現

主要是想通過復現一些簡單經典的gans模型來學習pytorch,之前 的 一直用tensorflow,聽說工業界pytorch用的居多,最近幾年pytorch也比tensorflow更被人看好,因此趁此機會學習一下。復現的文章有 1 最開始的gans 題目 generative adversaria...

SSD在pytorch下嘗試

conda create n pytorch python 3.7安裝pytorch conda install pytorch torchvision cudatoolkit 9.2 c pytorch太慢了,換 pip install torch torchvision啟用環境 git clon...