![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 61
dataat
这个作者很懒,什么都没留下…
展开
-
深度学习---自有图像数据集划分
当然 前提是将自有图像数据集已经按照分类进行了预处理,每个分类的图像作为一个单独的目录。要对自有图像数据集进行图像分类,首选需要将自有图像数据集划分为train和val(或者test)数据集。在执行了上述代码之后,实现了自有图像数据集的划分,然后就可以利用该数据集进行模型训练了。原创 2023-08-13 18:56:47 · 468 阅读 · 0 评论 -
gym基础程序调试
介绍gym基础程序运行过程中如何调试,发现问题和解决问题。最终支出gym的版本很重要。原创 2023-05-29 16:16:47 · 1156 阅读 · 0 评论 -
安装gym指定版本
笔者近期接触深度强化学习,想按照网上的相关资料实际操作和学习一番,而深度强化学习的实例中,很多都用到了gym这个库。有关这个gym库的介绍,笔者在此不再赘述,只简单说一下实际使用过程中遇到的问题。原创 2023-05-29 16:07:10 · 3116 阅读 · 1 评论 -
windows系统中安装目标检测平台detectron2
detectron2是Facebook研发并开源的目标检测平台,包含了大量业内最具代表性的目标检测、图像分割、关键点检测算法等。第二步: 根据python版本选择Jpype进行安装 https://www.lfd.uci.edu/~gohlke/pythonlibs/#jpype。但是笔者这里安装的时候报错了,忽略该问题,直接跳转到上面安装Detectron2,发现成功安装。笔者在执行第一行命令进行下载的时候,完全没有问题。但是第二行则直接报错,相关错误如下:。笔者经查询,解决方案如下:。原创 2023-04-21 09:23:18 · 1242 阅读 · 1 评论 -
pytorch利用resnet50提取图像特征
import osimport torchimport torch.nn as nnfrom torchvision import models,transformsfrom torch.autograd import Variableimport numpy as npfrom PIL import Imageimport timedevice=torch.device("cuda:3" if torch.cuda.is_available() else "cpu")class n.原创 2021-02-05 09:34:14 · 7373 阅读 · 6 评论 -
GPU显存不足,报错:RuntimeError: CUDA out of memory.
笔者近日利用pytorch框架对图像采用Resnet50模型提取特征的时候,利用到了GPU,发现报错如下:RuntimeError: CUDA out of memory. Tried to allocate 858.00 MiB (GPU 3; 10.92 GiB total capacity; 10.10 GiB already allocated; 150.69 MiB free; 10.13 GiB reserved in total by PyTorch)该程序运行所用GPU为1080TI原创 2021-02-04 14:11:32 · 3478 阅读 · 1 评论 -
报错:caffe2_detectron_ops.dll“ or one of its dependencies.的解决办法
在windows系统中安装了pytorch,结果运行最简单的测试程序时候仍然报错,大体信息如下: File "D:\Python38\lib\site-packages\torch\__init__.py", line 129, in <module> raise errOSError: [WinError 126] 找不到指定的模块。 Error loading "D:\Python38\lib\site-packages\torch\lib\caffe2_detectro.原创 2021-01-14 21:58:47 · 22265 阅读 · 38 评论 -
在Ubuntu中安装fastai
在深度学习领域,最受学生欢迎的MOOC课程平台有三个:Fast.ai、deeplearning.ai /Coursera和Udacity。Fastai作为其中之一,是一个课程平台,一个讨论社区,也是一个基于PyTorch的顶层框架,可以让新手快速的实现深度学习算法。主要是由fast.ai团队等人将相关重要论文中的算法在各种数据集上进行测试,如果测试效果较好,就会集成在库中,以便于用户可以方便使用,最终形成了该框架。今天本文简单介绍一下如何在Ubuntu系统中安装fastai。直接使用pip3原创 2020-08-28 13:55:33 · 311 阅读 · 0 评论 -
简单粗暴,上手修改原始神经网络模型的结构,以mobilenetv2为例
如题,想用mobilenetv2来训练自己的数据集,那肯定需要对原有的模型进细微的修改,至少需要把分类数给修改一下吧。要想细微修改网络模型,则需要首先知道网络模型的结构,以mobilenetv2为例,简单使用如下代码:from torchvision.models import mobilenet_v2import torch.nn as nnnum_classes=100net=mobilenet_v2().cuda()print(net)运行上述代码,显示mobilenetv2原创 2020-07-15 20:29:06 · 3876 阅读 · 0 评论 -
使用pytorch针对自己的数据集进行模型训练,并迁移至android app中,错误问题的原因以及解决之道
如题,前几天,笔者尝试了将pytorch 深度学习模型迁移至android系统中,并写了一篇小结:《Android移动终端应用APP中实现图像分类功能-----以pytorch为例》。在该文中,下载了原作者提供的resnet18模型并迁移至android app中,成功运行。至此,笔者甚为高兴,以为彻底解决了难题,可以随意迁移网络模型。孰料,将自己整理的图像数据集作为样本库进行迁移学习之后,迁移到android app中,直接一闪而过,还没来得及黑屏就退出了。真是一点情面都不给。在测试了很多次之后,笔者原创 2020-07-10 19:48:18 · 3392 阅读 · 4 评论 -
pytorch基于resnet18预训练模型用于自己的训练数据集进行迁移学习
本文记录利用resnet18预训练模型进行迁移学习,在自己的训练数据集上进行重新训练。相关代码重点部分分别介绍如下:model=torchvision.models.resnet18(pretrained=True)num_features=model.fc.in_featuresmodel.fc=nn.Linear(num_features,num_classes)model=model.to(device)说明:上述代码首先加载resnet18预训练模型,然后根据训练数据集中的分类数量原创 2020-07-08 20:55:54 · 14477 阅读 · 6 评论 -
pytorch加载模型时报错:AttributeError: Can‘t get attribute ‘MobileNet‘ on 「module ‘__main__‘
在使用pytorch加载模型时(具体命令为:torch.load("xxxxx.ckpt"),报错如下:D:\Python36\python.exe model2android.pyTraceback (most recent call last): File "model2android.py", line 3, in <module> model=torch.load("MobileNetv2.cpkt") File "D:\Python36\lib\site-pac原创 2020-07-07 11:15:16 · 16560 阅读 · 2 评论 -
pytorch加载训练数据集dataloader操作耗费时间太久,该如何解决?
笔者在使用pytorch加载训练数据进行模型训练的时候,发现数据加载需要耗费太多时间,该如何缩短数据加载的时间消耗呢?经过查询相关文档,总结实际操作过程如下:1、尽量将jpg等格式的文件保存为bmp文件,可以降低解码时间;2、dataloader函数中增加num_workers参数,该参数表示加载数据的线程数,建议设置为该系统中的CPU核心数,若CPU很强劲,而且内存很大,也可以考虑将该数值设置的更大一些。train_loader=torch.utils.data.DataLoader(dat原创 2020-07-06 14:36:04 · 9499 阅读 · 2 评论 -
运行pytorch框架下的图像分类训练程序,出现cuda out of memory,解决方法探索
今天利用python和pytorch编写图像分类训练程序,好不容易噼里啪啦敲完键盘,运行之。。。。。,结果突然报错(RuntimeError cuda out of memory),使笔者大失所望,具体信息如下:原创 2020-07-06 09:40:15 · 3430 阅读 · 0 评论 -
python程序中rmdir删除非空目录,报错的解决方法
今天在python中删除目录dirname,调用了os.rmdir(dirname),因为dirname目录中含有内容,为非空目录,因而导致报错,具体信息如下:OSError: [Error 39] Directory not empty: '/home/xxx/dirname'建议改为调用shutil,具体代码如下:shutil.rmtree("/home/xxx/dirname")顺利运行通过...原创 2020-07-03 14:55:04 · 5975 阅读 · 0 评论 -
Android移动终端应用APP中实现图像分类功能-----以pytorch为例
笔者想在Android版本的App中实现图像分类的功能,查找相关文档之后,决定基于Pytorch实现,具体可以参考https://pytorch.org/mobile/android/。本文详细记录笔者进行实际测试的过程。1、利用预训练模型进行配置和测试模型转换:需要将预训练模型(例如Resnet18)进行序列化,相关代码如下:import torchimport torchvisionmodel = torchvision.models.resnet18(pretrained=True原创 2020-07-02 11:09:41 · 1711 阅读 · 0 评论 -
安装测试YOLOv5
不久前,刚刚发布了YOLOv5版本的源代码,并且可以在github网站下下载,笔者今天下载下来试用了一下,对于遇到的一些问题做个记录,希望能让大家少走弯路。YOLOv5网址:https://github.com/ultralytics/yolov5访问该网址,在Ubuntu16.04的系统中进行安装:pip install -U -r requirements.txt能够顺利安装完成。随后按照提示,运行如下命令进行测试:$ python detect.py --source file原创 2020-06-28 13:57:57 · 3591 阅读 · 3 评论 -
YOLO系列目标检测算法简述
说起目标检测算法,就不得不提到YOLO,尤其是今年YOLO版本有了新的发布。1、YOLOv3目前在YOLO官网上(https://pjreddie.com/darknet/yolo/),YOLO最新版本是v3,关于YOLOv3的使用key参考官网,网络中的相关文章也很多,可以自行搜索了解。不过这里有必要澄清一个概念,就是YOLO与darknet的关系:darknet是YOLO作者自己写的一个深度学习框架,后来在YOLO9000中提出了由ResNet改进而来的19层卷积网络,称之为Darknet-原创 2020-06-26 15:50:17 · 440 阅读 · 0 评论 -
线性回归的pytorch代码
使用pytorch实现的线性回归,闲言少叙,直接上代码,客官请看:import torchimport torch.nn as nnimport numpy as npimport matplotlib.pyplot as plt#设置相关参数input_size=1output_size=1num_epochs=60learning_rate=0.001#导入训练数据集x_train=np.array([[3.3],[4.4],[5.5],[6.71],[6.93],[4.原创 2020-06-23 16:25:08 · 303 阅读 · 0 评论