shell$ git clone https://github.com/thtrieu/darkflow.git
下载darknet,并按照Reanme.me安装。
之后将自己的数据(VOC2007)放入scripts文件夹下。打开voc_label.py。
修改类别:我是三分类,并且只有VOC2007的数据,那么
sets=[('2007', 'train'), ('2007', 'val'), ('2007', 'test')]
classes = ["normal", "inclined", "benttodesk"]
改最后两行为:
os.system("cat 2007_train.txt 2007_val.txt > train.txt")
os.system("cat 2007_train.txt 2007_val.txt 2007_test.txt > train.all.txt")
运行voc_label.py,会生成一些.txt文件用来训练。