摘要:
mmdetection 绘制PR曲线 发现直接使用matplotlib绘制曲线在修改图片上一些细节是比较麻烦,因此我决定使用Excel来绘制PR曲线, 如果你想直接画PR曲线,请直接看第二段代码 导出Excel import os import mmcv import numpy as np imp 阅读全文
posted @ 2022-03-27 14:30
Ghlerrix
阅读(341)
评论(0)
推荐(0)
摘要:
使用Python在终端打印表格 from prettytable import PrettyTable table = PrettyTable(['Title1', 'Title2', 'Title3']) table.add_row([1, 2, 3]) table.add_row([4, 5, 阅读全文
posted @ 2022-03-27 14:26
Ghlerrix
阅读(25)
评论(0)
推荐(0)
摘要:
自动从训练结果中获取最高的mAP所对应的epoch。 <code>import json import os ''' :param work_dir 训练结果目录 :return 最好的map对应的epoch路径 ''' def getBestEpoch(work_dir): global maxE 阅读全文
posted @ 2022-03-27 14:25
Ghlerrix
阅读(38)
评论(0)
推荐(0)
摘要:
训练出现nan 在使用MMDetection训练模型时,发现打印信息中出现了很多nan。现象是,loss在正常训练下降的过程中,突然变为nan。 梯度裁减 在模型配置中加上grad_clip: optimizer_config = dict(grad_clip=dict(max_norm=35, n 阅读全文
posted @ 2022-03-27 14:22
Ghlerrix
阅读(368)
评论(0)
推荐(0)
摘要:
PyTorch 广播机制 定义 PyTorch的tensor参数可以自动扩展其大小。一般的是小一点的会变大,来满足运算需求。 规则 满足一下情况的tensor是可以广播的。 至少有一个维度两个tensor维度相等维度不等,其中一个为1维度不等,其中一个维度不存在 计算过程规则。 维度不同,小维度的增 阅读全文
posted @ 2022-03-27 14:17
Ghlerrix
阅读(44)
评论(0)
推荐(0)
摘要:
Pytorch 实现线性回归 import torch from torch.utils import data from torch import nn # 合成数据 def synthetic_data(w, b, num_examples): """y = Xw + b + zs""" X = 阅读全文
posted @ 2022-03-27 14:15
Ghlerrix
阅读(19)
评论(0)
推荐(0)
浙公网安备 33010602011771号