语义分割实践思考记录(个人备忘录)

一、任务管理器、NVDIA的GPU利用率显示12

若需要在任务管理器中查看基于Pytorch框架的GPU利用率,那么,我们需要将监控面板监测内容调整为cuda。图一(左)即为英伟达命令行工具面板。
图一 英伟达GPU使用率监控

二、基于混淆矩阵的语义分割评价指标计算

在语义分割数据集标签制作的过程中,我们通常根据分类类别数N来以0 --- (N-1)的整型数字标记不同的分类目标。通过图二可以看出,在相同预测模型且相同数据量的情况下,方法二评价指标计算效率高(较方法一节约了一半时间),同时,充分有效地利用了上述标签标记的机制。

方法二的混淆矩阵代码编写简述:首先,根据语义分割任务分类类别数N创建N×N(N行N列)的零矩阵,然后利用pytorch中的view()函数将单幅预测结果图及标签图分别展平为一行,再者利用Python的zip()函数组合对应像素的混淆矩阵坐标,最后根据混淆矩阵坐标不断循环加一即可。若混淆矩阵坐标为(预测值,标签值),则行号为预测结果,列号为真值。
图二 混淆矩阵生成与精度指标计算

参考资料:

1NVIDIA查看CPU、内存、GPU、DLA使用情况_宗而研之的博客-CSDN博客

2Tensorflow模型GPU使用率低的问题_模流分析软件不占gpu_CooL截击的博客-CSDN博客

相关推荐
hhzz1 小时前
《深度学习框架PyTorch入门与实践》系列:01-PyTorch入门与环境搭建之从安装到第一个神经网络
人工智能·pytorch·神经网络
从零开始学习人工智能2 小时前
PyTorch 踩坑:Input type \(Half\) and bias type \(float\) 类型不匹配终极解决方案
人工智能·pytorch·python
盼小辉丶4 小时前
PyTorch计算机视觉(5)——生成对抗网络(Generative Adversarial Network,GAN)
pytorch·生成对抗网络·计算机视觉
一碗白开水一1 天前
入门实践工程四:基于 PyTorch 的手写数字识别(MNIST 图像分类)|附:环境依赖环境及工程源码
人工智能·pytorch·分类
雪的季节1 天前
不安装YOLO只安装 PyTorch,加载已有yolo数据,从无到有创建模型训练数据并加载使用(重要)
人工智能·pytorch·yolo
一次旅行1 天前
DPO/KTO/ORPO深度对比:三大LLM偏好对齐数学推导+完整PyTorch实现+落地选型指南
人工智能·pytorch·python
淼澄研学1 天前
PyTorch核心实操:从张量计算到混合精度训练的5个关键步骤
人工智能·pytorch·python
淼澄研学1 天前
PyTorch 2.0 核心机制解析与5个实操方法
人工智能·pytorch·python
盼小辉丶1 天前
PyTorch计算机视觉(4)——迁移学习(Transfer Learning)详解与实现
pytorch·计算机视觉·迁移学习
是上好佳佳佳呀2 天前
【深度学习|Day02】PyTorch 深度学习笔记(下):张量运算与自动微分
pytorch·笔记·深度学习