August李

深度学习

C++(28) python(3) 个人总结(2) 嵌入式开发(3) 开发环境(3) 数字图像处理(15) 题解(1)

/ 注册

全部文章 / 深度学习（共15篇）

创建用于DataLoader的pytorch数据集

暂时介绍 image-mask型数据集, 以人手分割数据集 EGTEA Gaze+ 为例. 准备数据文件夹需要将Image和Mask分开存放, 对应文件的文件名必须保持一致. 提醒: Mask 图像一般为 png 单通道 EGTEA Gaze+ 数据集下载解压后即得到如下的目录, 无...

2020-07-13

0 730

PyTorch 报错处理汇总(包含CUDA)

Error 汇总 CUDA out of memory 原因: GPU内存溢出解决: 减小 BATCH_SIZE CUDA error: device-side assert triggered 原因: 实际输入的label类别或Mask图像的像素类别超过了设定的类别数 NUM_...

2020-07-13

0 2796

分阶段训练 Stagewise Training 简介

参考论文: 2015: Stage-wise training: An improved feature learning strategy for deep models 2018 Why does stagewise training accelerate convergence o...

2020-07-13

0 1352

numpy转PIL.Image: 处理Mask图像为单通道的彩色/灰度图colormap.png

文章目录从分割数据集说起 8位彩色图 8位灰度图 1位二值图什么是图像位深 PNG图像格式简介如何处理Mask图像 PIL库相关函数从分割数据集说起 8位彩色图下图是 NYU数据集里的一张Mask图像, 显示彩图, 但实...

2020-07-13

0 4631

在C++ OpenCV 环境中部署PyTorch模型

文章目录导出 TorchScript 模型设置LibTorch环境示例代码总结和参考本文环境： win10 + vs2019 + opencv 4 导出 TorchScript 模型有两种方式，可以自己试一下，而且很多开源模型已经提供了转换脚本，...

2020-07-13

0 1661