摘要：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维度不等，其中一个维度不存在 计算过程规则。 维度不同，小维度的增 阅读全文

摘要：Pytorch 实现线性回归 import torch from torch.utils import data from torch import nn # 合成数据 def synthetic_data(w, b, num_examples): """y = Xw + b + zs""" X = 阅读全文