小白初学pytorch

原创

已于 2022-08-29 10:27:04 修改 · 947 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

于 2022-08-28 18:03:48 首次发布

这篇博客记录了作者学习PyTorch的过程，包括跟随B站up主牛_奶奶的教程跑通首个神经网络，理解如何优雅地进行图像分类，以及深入学习Dataset数据集的CSV和VOC格式。文中提到了Kaggle数据集资源，并对CSV和VOC数据集的处理进行了说明。

跟着b站up主：牛_奶奶，学习pytorch，学习笔记记录中。。。
up主源码地址：https://github.com/ssssww0905/-PyTorch-
10个适合数据人练手的Kaggle数据集：https://zhuanlan.zhihu.com/p/459829584
Kaggle dog_vs_cat 数据集下载：https://www.microsoft.com/en-us/download/confirmation.aspx?id=54765

1_手把手教你跑通第一个神经网络

另外一篇非常详细讲“用PyTorch实现MNIST手写数字识别”的文章：
https://zhuanlan.zhihu.com/p/137571225

2_如何让网络优雅地分类猫猫狗狗

pip install tensorboard
tensorboard --logdir=runs

5_Dataset数据集实例讲解，CSV数据&VOC格式

5.1 数据集介绍

CSV数据（案例：肿瘤分类数据集）
expression.csv 每一列是一个样本，每个样本有5万多个基因表达。
annotation.csv 每一行是一个样本，标签只有2个值（Tumor或Normal）。
每个样本都有相同个数的特征表达，且DataLoader默认的collate_fn会自动将int转为Tensor类型，不需要重定义。

VOC数据（Visual Object Classes Challenge）（案例：火焰数据集）
简单来说，就是一张图片对应一个xml文件
图片中，原点为左上角，xmin,ymin对应图片左上角坐标，xmax,ymax对应图片右下角坐标
每张图片大小不一定相同，还需要重定义DataLoader默认的collate_fn。