RCNN-keras所需样本集。The train_alexnet.py uses the 17flowers image folder with the train_list.txt file to perform the pre-training of the Alexnet. This is the file that need to be run first.
After it generates the model_save.model file, you can run the f
如下所示:
import os
import os.path
rootdir = "./pic_data"
file_object = open('train_list.txt','w')
for parent,dirnames,filenames in os.walk(rootdir):
for filename in filenames:
print filename
file_object.write(filename+ '\n')
file_object.close()
在做深