-
修改显卡:cfgs.py文件 第15行 GPU_GROUP=“1”
查看显卡使用情况: nvidia-smi -
生成tfrecord文件:
-a: 改cfgs.py
第9行 NET_NAME=‘resnet_v1_50’ 作者用的是101
第8行 VERSION=‘RRPN_VOCdevkit2007(MLT)_v1’
第64行 DATASET_NAME=‘VOCdevkit2007(MLT)’
CLASS_NUM改为1 试一试
-b: 在/libs/label_dict.py里面添加dataset
elif cfgs.DATASET_NAME=='VOCdevkit2007(MLT)':
NAME_LABEL_MAP={
'background' : 0