摘要:
mmdetection 绘制PR曲线 发现直接使用matplotlib绘制曲线在修改图片上一些细节是比较麻烦,因此我决定使用Excel来绘制PR曲线, 如果你想直接画PR曲线,请直接看第二段代码 导出Excel import os import mmcv import numpy as np imp
阅读全文
摘要:
使用Python在终端打印表格 from prettytable import PrettyTable table = PrettyTable(['Title1', 'Title2', 'Title3']) table.add_row([1, 2, 3]) table.add_row([4, 5,
阅读全文
摘要:
自动从训练结果中获取最高的mAP所对应的epoch。 <code>import json import os ''' :param work_dir 训练结果目录 :return 最好的map对应的epoch路径 ''' def getBestEpoch(work_dir): global maxE
阅读全文
摘要:
训练出现nan 在使用MMDetection训练模型时,发现打印信息中出现了很多nan。现象是,loss在正常训练下降的过程中,突然变为nan。 梯度裁减 在模型配置中加上grad_clip: optimizer_config = dict(grad_clip=dict(max_norm=35, n
阅读全文
摘要:
PyTorch 广播机制 定义 PyTorch的tensor参数可以自动扩展其大小。一般的是小一点的会变大,来满足运算需求。 规则 满足一下情况的tensor是可以广播的。 至少有一个维度两个tensor维度相等维度不等,其中一个为1维度不等,其中一个维度不存在 计算过程规则。 维度不同,小维度的增
阅读全文