本人是第一次完整的制作了一个自己的数据集,因为各种都是从零开始,所以比较详细,适合和我一样的小白阅读。如果哪里写的有问题,欢迎指正。
一、首先,找到数据集需要的图片,或者用ps制作。这里说明一下最好做成jpg 格式的。因为在VOCLike函数读取的时候默认的是jpg格式。如果你不想改也没关系,可以修改VOCDetection这个函数,修改为png或其他格式。具体如下;
1.在detection.py中找到VOCDetection这个函数
2.修改下图中的{}.jpg的部分为你图片的格式
也可以直接将图片格式修改成jpg的,下面贴一段代码:
import os
import string
dirName = "/home/anzhuo/pic/" #这里改成自己的图片所在路径
li=os.listdir(dirName)
for filename in li:
newname = filename
newname = newname.split(".")
if newname[-1]=="png": #这里是你图片的原格式的后缀
newname[-1]="jpg"
newname = str.join(".",newname) #这里要用str.join
filename = dirName+filename
newname = dirName+newname
os.rename(filename,newname)
print