目标检测，关于Gluon的SSD跑自己的数据-只需要lst，不需要rec

关于SSD代码的有些问题可以看我的上一篇的部分代码解释 关于gluon中Object Detection 中SSD模型的笔记

当然还有一些地方没有写清楚，主要就是MultiBoxTarget和MultiBoxDetection两个函数。但是大致可以理解为第一个函数是在算anchor与label之间的有没有交叉的关系，毕竟是个多尺度的网络。第二个函数就是在得到网络计算出的offset基础上和已知的anchor基础上，生成网络预测的实际bouding box。仅供参考。后续有空会看一下源码或者更多的资料。

这里只要说一下数据准备阶段。

可以看到代码主要用了一个函数ImageDetIter，上一篇代码讲解里面，我也提到了这个问题“多目标怎么办”。上面我的代码只跑了单目标的数据集，然后采用的ndarray的iterator。我之前一直也很迷惑为什么后面label要reshap成3*5。

然后我就去看了这个函数，总结来说就是就是你只需要一个lst文件就可以生成一个iterator了，里面还会根据你的不同的目标，去reshape成不同的大小（空的补足-1）。

如何生成不同的list了，上面也给了说明，但是我觉得很不清楚啊，这个是空格呢，还是什么呢。反正就是我经过测试几次之后发现，每个符号之间隔‘\t’就可以了。每个字符的意思就是 xmin, ymin, xmax, ymax，这四个也应该是（ xmin\t ymin\t xmax\t ymax）

给你们一个样例，自己写这个一个.lst的文件就可以了

'1700\t4\t5\t804\t734\t0\t0.553482587065\t0.0940054495913\t0.813432835821\t0.298365122616\t/root/workspace/data/DeepBC/DeepBC_3D/dataset/processed/train/15_191.jpg'

1700 是编号，第1700张图片

4 据说一定要4， 他的意思是(2 + length of extra header)， 2是 4和5，length of extra header是804，734，这里是图片的大小

5 就是每个label的组成，categories，xmin, ymin, xmax, ymax

后面的就是正常的坐标了

最后是这个label对应的图片，请写绝对路径

然后验证一下自己的lst是不是写对了

import mxnet as mx
data_iter = mx.image.ImageDetIter(batch_size=4, data_shape=(3,804, 734),path_imglist='./train.lst')

data_iter.reset()
for data in data_iter:
    d = data.data[0]
    l = data.label[0]
    print(d.shape)
    print(l.shape)
    break # 看一组就可以了

然后也可以看一下自己的图片的label对不对

import numpy as np
import matplotlib.pyplot as plt
idx = 0
img = d[idx].asnumpy()  # grab the first image, convert to numpy array
img = img.transpose((1, 2, 0))  # we want channel to be the last dimension #img += np.array([123, 117, 104])
img = img.astype(np.uint8)  # use uint8 (0-255)
for ll in l[idx].asnumpy():
    if ll[0]<0: #-1是补充的，不是label
        break
    else:
        print(ll)
        xmin= int(ll[1]*734) #因为我的x的shape是734
        ymin= int(ll[2]*804) #因为我的y的shape是804
        xmax= int(ll[3]*734) 
        ymax = int(ll[4]*804) 
        rect = plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin, fill=False, edgecolor=(1, 0, 0), linewidth=1)
        plt.gca().add_patch(rect)
        break

plt.imshow(img,'gray')# 我的是灰度图，如果你们是彩色图也可以做别的操作
plt.show()

你可以plot一下就知道自己传给网络的data到底是不是正确的了

如果有错误就请指出~