SSD.pytorch学习

1、如果只有一类,比如做人脸检测、行人检测

只有一个类名要写('类名',),逗号千万不能忘!

https://www.freesion.com/article/7565405327/

通过查阅,有以下这几种写法,不知道哪个是对的。。 

假设类名叫tanju

VOC_CLASSES = [('tanju')]

VOC_CLASSES = (  # always index 0
     'tanju','background',)

VOC_CLASSES = (  # always index 0
    'tanju',)

VOC_CLASSES = ('__background__',# always index 0
     'tanju')

如果是这种写法,会报错

VOC_CLASSES = ('tanju')

官方代码只有多类的写法:

VOC_CLASSES = (  # always index 0
    'aeroplane', 'bicycle', 'bird', 'boat',
    'bottle', 'bus', 'car', 'cat', 'chair',
    'cow', 'diningtable', 'dog', 'horse',
    'motorbike', 'person', 'pottedplant',
    'sheep', 'sofa', 'train', 'tvmonitor')

 

你可能感兴趣的:(小目标检测)