SSD在pytorch下嘗試

conda create -n pytorch python=3.7

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

pip install torch torchvision

git clone

但是發現資料集在主目錄上，還是沒法使用，建立了個連線直接跳轉

ln -s /home/king/data/coco /home/king/ssd.pytorch/data

ln -s /home/king/data/vocdevkit /home/king/ssd.pytorch/data

#專案根目錄下

mkdir weights
cd weights
wget 

感謝這個教程：

直接按照4裡面的修改:

修改1

loc_loss += loss_l.data

conf_loss += loss_c.data
if iteration % 10 == 0:
print('timer: %.4f sec.' % (t1 - t0))
#loss.data[0]
print('iter ' + repr(iteration) + ' || loss: %.4f ||' % (loss.data), end=' ')

try: 

images,targets = next(batch_iterator)
except stopiteration:
batch_iterator = iter(data_loader)
images, targets = next(batch_iterator)

我的1060我一直修改學習率到0.0000005

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

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

基於pytorch的SSD復現筆記

復現的過程按照github的步驟進行 但注意有幾個坑 sudo apt install curl兩個命令的使用參考 2.cv2的安裝 詳細參考我的另一篇部落格 cv2的安裝包名字不叫cv2,而是叫opencv python 3.pytorch版本需要為0.4.1,python版本為python3.6...

目標檢測演算法SSD在Ubuntu CPU下執行

定位到 caffe ssd examples ssd ssd detect.py，修改如下 源 make sure that caffe is on the python path caffe root 改為 我把caffe ssd資料夾放在了主目錄home下 make sure that caff...