pytorch
(12)将 yolov5 内置模型导出为 pytorch mobile 格式
## Google Colab 地址
使用 Colab 能利用 Google 免费提供的 GPU 资源,快速进行模型训练。比在我的弱鸡笔记本上训练,能节省大量时间。
在 Colab 上新建一个项目
https://colab.
yolov5 模型导入 Android 后编译的 apk 文件体积巨大
参考前文,[用 yolov5 的 pytorch torchscript lite 模型开发了一个多物体检测的 Android APP](/yolov5-pytorch-import-android-mobile-model-realize
yolov5 pytorch mobile 模型导入 Android,实现图片目标检测
由于 yolov5 相关的 TensorFlow Lite 示例代码太少,而且都是旧版本的示例,导致我完全不知道怎么上手。
所以转投 pytorch mobile 的方案,主要是 facebook 官方的示例代码非常完整,且清晰明了,容易
yolo v5 导出 Android 手机上可以运行的 TFLite 模型
TFLite 是 TensorFlow Lite 的缩写。TFLite 在 Android 上的运行速度更快,因为它专门针对移动设备进行了优化。但是,最后我发现我实在找不到在 Android 上执行 yolov5 TFLite 模型的示例,
YOLO v5 PyTorch 训练数据标注文件 YAML 格式
从 roboflow 标注数据之后,导出的 YOLO v5 PyTorch 配置文件 YAML 格式 (TXT annotations and YAML config used with YOLOv5),里面的文件路径非常不友好,每次都需要
如何判断当前电脑上的 pytorch 是否支持 CUDA
## CUDA
CUDA 是 Nvidia GPU 提供的软件 API。
## 查看显卡型号
Windows 11,我的电脑 - 右键属性 - 设备管理器 - 显示适配器
例如,我这个 Dell 台式机是:
- N
新建一个神经网络,方式一,基于 torch.nn.Sequential
## 构建神经网络的两种方式
PyTorch 有两种构建神经网络的方式:
- 简单粗暴式:使用 torch.nn.Sequential 类,初始化时,传入一系列操作 (operation)
- 通用式:定义 torch.nn.Mo
神经网络 (neural network) 概念
在 PyTorch 的教程及书籍中,经常会看到神经网络这个词。
英文是:neural network。
neural 是个形容词,神经的,神经系统的。中文发音:呢若 / 牛若。
代码中也常见到
```
torch.
PyTorch 基础概念 - 数据集 Dataset
## Datasets and Dataloaders
- Dataset (torch.utils.data.Dataset) 存储了样本及其对应的标签。
- DataLoader (torch.utils.data.DataLoad
PyTorch 基础概念 - 张量 Tensors
无论是 PyTorch 还是 TensorFlow,我们首先都会遇到一个基础概念,张量 (Tensor)。
## 张量 (Tensor) 的作用
模型 (Model) 的输入和输出都是用张量 (Tensor) 表示的。
#
上一页
下一页