您的當(dāng)前位置： > 關(guān)注 > > 正文

新消息丨windows10運(yùn)行ssd訓(xùn)練自己的數(shù)據(jù)集 pytorch1.8.0+cuda111的數(shù)據(jù)集

來源：CSDN 時間：2023-03-31 07:51:34

運(yùn)行環(huán)境：

1、win10 2、pytorch 1.8.0+cuda 111（1.9.0會報錯） 3、pycharm

（一）數(shù)據(jù)集準(zhǔn)備階段

(資料圖)

第一步：新建VOC2007文件夾，然后再在里面新建三個文件ImageSets–>main JPEGImages（存放照片），Annotations（存放xml文件）。第二步：對自己的照片進(jìn)行標(biāo)注，工具為labelimg，這個就不多贅述了

（二）ssd代碼準(zhǔn)備階段

第一步：再github下載star最多的ssd源代碼https://github.com/amdegroot/ssd.pytorch 第二步：下載權(quán)重文件VGG16 https://s3.amazonaws.com/amdegroot-models/vgg16_reducedfc.pth 第三步：解壓ssd文件，解壓之后在打開SSD/data文件夾，在這之下新建一個VOCdevkit文件夾第四步：將自己的VOC2007數(shù)據(jù)集放到VOCdevkit文件夾中第五步：在根目錄下新建一個weights文件夾，將VGG16放進(jìn)去

（三）sdd代碼修改階段

第一步：將文件導(dǎo)入pycharm中打開第二步：打開data/init.py：將第三行的from .coco import COCODetection, COCOAnnotationTransform, COCO_CLASSES, COCO_ROOT, get_label_map 注釋掉。因?yàn)槲覀儾皇褂胏oco數(shù)據(jù)集，而是使用自己訓(xùn)練的。第三步：打開data/config.py：第十五行的 num_classes改成自己設(shè)定的類別數(shù)+1，所以我這里是2 第四步：打開data/voc0712.py:

第二十行的VOC_CLASSES =改成自己的類別名，我的是tree；第二十三行的代碼改成VOC_ROOT = ‘./data/VOCdevkit/’；第93行改為image_sets=[(‘2007’, ‘trainval’)]。

第五步：打開layers/modules/multibox_loss.py：在第97行的loss_c[pos] = 0的上面加上一句loss_c = loss_c.view(num, -1)，在第115行N = num_pos.data.sum()的后面加上.double,繼續(xù)在下面添加

這兩句代碼

第六步：打開ssd.py程序：把這里面所有的num_classes的數(shù)量都改為類別數(shù)+1，我的都是6。第七步：打開train.py程序：搜索這里面的data[0]，把這里面的[0]都刪掉，只保留data。第84行的這兩行#if args.dataset_root == COCO_ROOT: # parser.error(‘Must specify dataset if specifying dataset_root’)注釋掉，因?yàn)槲覀儾挥肅OCO。第165行的images, targets = next(batch_iterator)改成

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

第198行的5000我改成500；迭代500次保存一次文件，大家可改可不改。第200行權(quán)重的名稱自己隨意，我改的是weights/ssd300_voc_。

（四）開始訓(xùn)練

** 第一步：修改train.py中的batch-size，我的是rtx3060（6g），我就修改為24

第二步：開始訓(xùn)練 python train.py （這個loss看著就不正常）

（五）測試

（六）遇到的問題

** 訓(xùn)練的時候loss=“nan”，修改train.py中的這個，修改為1e-4

責(zé)任編輯：

標(biāo)簽：

上一篇：每日快訊!2019年運(yùn)城高考狀元名單公布：運(yùn)城文理科狀元是誰？
下一篇：最后一頁

相關(guān)推薦：

精彩放送：