课程使用program_aiot资源, 程序支持 pascal voc 格式的 detection 任务,wider_VOC_small 文件夹为预准备的小数据集,该数据集为 pascal voc 格式标签,数据节选自 wider face,共 200 张人脸图像,其中训练集 150 张,验证集 50 张;
资料库中给出了预训练模型 pre_train_weight,搭建时 pre_model 文件夹中应有预训练模型,dataset 文件夹中有数据集 wider_VOC_small,并更改配置文件 configuration.py,分别配置预训练模型路径、数据集路径、标签。
通过 voc 数据集生成 txt 格式数据,执行 python write_voc_to_txt.py,生成 txt 文件数据,存放为 program_aiot/data_process/data.txt。
执行 python train_from_scratch.py ,启动训练 , 训 练 完 的 模 型 保 存 在program_aiot/saved_model/目录下,完成训练需要较长时间。
实验过程中遇到找不到tensorflow模块的情况,搜寻后解决方法如下:
再次运行训练代码