Pytorch初学

创建虚拟环境

python控制台,jupyter notebook,python文件运行的差异,后续结合使用三者。

jupter主要可以对代码进行分割单独运行,主要做一些探索性工作。

数据集的常见存储模式

1、以标签命名图像。

2、单独存储图像的地址。

加载数据集:

dataset 和dataloader类,dataset

dataset获取每一个数据及其label,以及数据的个数。

你可以把 Dataset 想象成一个超级收纳箱,里面存放着机器学习模型要学习的所有数据。这些数据可以是各种各样的,比如图片、文字、数字等。它把这些数据按照一定的规则整理好,方便模型随时取用。

在命令行获取单个图像的数据集:

首先获取图像地址(相对地址较好)。

获取所有图像地址,导入os,创建所有图片的列表。

看到每个变量对应的属性(控制台的优点)

得到图像集列表的逻辑如下,将根目录与本图形数据分开:

复制代码
 root_dir=""
    lable_dir=""
    path=os.path.join(root_dir,lable_dir)
    img_path=os.listdir(path)

TensorBoard是个"魔法工具":

模型可视化

  • 展示计算图:能把模型内部结构清晰呈现,看清各层连接与数据流向,助你明白模型构造。

  • 查看模型参数:以图展示参数变化,让你知晓模型学习时参数的调整情况。

训练过程监控

  • 损失函数可视化:将损失函数绘成曲线,依曲线走势判断模型训练好坏,便于调整方法。

  • 评估指标展示:展示准确率等指标变化,方便对比选最优模型。

数据可视化

  • 输入数据可视化:可展示输入数据样子,助你了解模型学习的数据情况。

  • 特征可视化:呈现模型学习时关注的数据特点,利于理解其学习和决策依据。

打开端口

相关推荐
德育处主任3 分钟前
『OpenCV-Python』配合 Matplotlib 显示图像
后端·python·opencv
呆头鹅AI工作室13 分钟前
[2025CVPR-图象分类方向]CATANet:用于轻量级图像超分辨率的高效内容感知标记聚合
图像处理·人工智能·深度学习·目标检测·机器学习·计算机视觉·分类
向左转, 向右走ˉ19 分钟前
为什么分类任务偏爱交叉熵?MSE 为何折戟?
人工智能·深度学习·算法·机器学习·分类·数据挖掘
抓个马尾女孩22 分钟前
什么是熵、交叉熵、相对熵(KL散度)
人工智能·机器学习
张较瘦_1 小时前
[论文阅读] 人工智能 | 机器学习工作流的“救星”:数据虚拟化服务如何解决数据管理难题?
论文阅读·人工智能·机器学习
蓝卓工业操作系统2 小时前
天铭科技×蓝卓 | “1+2+N”打造AI驱动的汽车零部件行业智能工厂
人工智能·科技·汽车
zzywxc7872 小时前
编程算法在金融、医疗、教育、制造业等领域的落地案例
人工智能·算法·金融·自动化·copilot·ai编程
zzywxc7872 小时前
编程算法在金融、医疗、教育、制造业的落地应用。
人工智能·深度学习·算法·机器学习·金融·架构·开源
修一呀2 小时前
【数据标注】详解使用 Labelimg 进行数据标注的 Conda 环境搭建与操作流程
人工智能·conda
白熊1886 小时前
【大模型LLM】梯度累积(Gradient Accumulation)原理详解
人工智能·大模型·llm