faster-rcnn模型评估
1.本文使用VOC格式进行评估。
2.如果在训练前已经运行过voc_annotation.py文件,代码会自动将数据集划分成训练集、验证集和测试集。如果想要修改测试集的比例,可以修改voc_annotation.py文件下的trainval_percent。trainval_percent用于指定(训练集+验证集)与测试集的比例,默认情况下 (训练集+验证集):测试集 = 9:1。train_percent用于指定(训练集+验证集)中训练集与验证集的比例,默认情况下 训练集:验证集 = 9:1。
利用voc_annotation.py划分测试集后,前往get_map.py文件修改classes_path,classes_path用于指向检测类别所对应的txt,这个txt和训练时的txt一样。评估自己的数据集必须要修改。
3.在frcnn.py里面修改model_path以及classes_path。model_path指向训练好的权值文件,在logs文件夹里。classes_path指向检测类别所对应的txt。
4.运行get_map.py即可获得评估结果,评估结果会保存在map_out文件夹中。

5.评估结果









ps:FigureCanvasTkAgg' object has no attribute 'set_window_title'将该处代码改为fig.canvas.manager.set_window_title即可

浙公网安备 33010602011771号