PyTorch 目标检测(五)
SSD实战 1. 先从跑别人的代码开始 https://github.com/amdegroot/ssd.pytorch 下载代码,markdown文件有教程,但是代码有一些问题 问题汇总 用的是VOC2012训练集 删除voc0712.py __init__函数中imageset参数里07的信息 运行Train.py RuntimeError: CUDA out of memory. Tried to allocate 176.00 MiB (GPU 0; 2.00 GiB total capacity; 1.23 GiB already allocated; 107.80 MiB free; 1.24 GiB reserved in total by PyTorch) 电脑太差,降低batch_size IndexError: The shape of the mask [8, 8732] at index 0 does not match the shape of the indexed tensor [69856, 1] at index 0 ssd.pytorch\layers\modules\multibox_loss.py中loss 维度不对应 layers/modules/multibox_loss.py 第97、98行调换 loss_c = loss_c.view