深度学习
xgbm_k
这个作者很懒,什么都没留下…
展开
-
使用docker执行超分辨率super resolution项目neural-enhance
最近在GitHub看到了一个super resolution的项目neural-enhance。作者提供了docker镜像供大家下载使用。作者提供的是在linux下使用该镜像的方法,我的操作系统是Win10。这里写一下我是如何下载镜像并使用该项目的。首先,下载并安装docker desktop for windows。其次,要从dock hub上下载作者的镜像。国内直接从dock hub上下载镜像非常慢,所以要设置一下,从国内的镜像站下载。按照下图所示在registry-mirrors中添加国内镜像站。原创 2020-09-17 11:44:16 · 548 阅读 · 1 评论 -
Pytorch nn.Transformer示例代码
Pytorch官方Transformer教程没有使用nn.Transformer,仅仅使用了TransformerEncoderLayer。甚至连TransformerDecoderLayer都没有用。我找到了一个nn.Transformer示例代码,自己改了改放在github上:https://github.com/Kenneth111/TransformerDemo有兴趣的可以参考一下。...原创 2020-03-10 11:23:49 · 5432 阅读 · 5 评论 -
用卷积神经网络(CNN)识别文字
好久没有写博客了,趁最近比较闲来发一篇。有个项目:从计算机屏幕上截图识别截图上的文字早先识别的方法是:对比度颜色边缘检测问题是:有些图片会被识别为文字。前几天闲着就试了一下用卷积神经网络(CNN)来实现这个功能。生成训练集、测试集和验证集。首先,拿到一幅带有文字的计算机画面,用工具找到文字区域的起始和终止坐标,再用工具找到图片区域的起始和终止坐标。用下面的代码生成训练...原创 2018-09-20 10:17:45 · 15895 阅读 · 0 评论 -
超分辨率生成对抗模型SRGAN
最近想把低质量的图片还原成高质量的图片,因此又回到了single image super resolution模型。一直想找一些简单的模型把问题搞定,但试了两天效果都不好。昨天又看到了SRGAN。这篇论文早些时候看过没有特别看懂,当时也是为了找简单的模型应用在工程上,就放过去了。昨天有时间好好看了一下。感觉还是有些收获的。论文中提出了一个观点:用MSE(也包括了psnr)作为loss得到的超分...原创 2018-09-28 14:16:11 · 2615 阅读 · 0 评论 -
Pytorch官方文档中练习的答案
最近在看PyTorch的官方教程。在官方教程中特别是NLP中有些练习没有给出答案。我在看教程时顺手把这些练习给做了,代码放在这里sequence_models_tutorial.py对应Augmenting the LSTM part-of-speech tagger with character-level features最后的练习。word_embeddings_tutorial.py对...原创 2018-11-12 10:32:21 · 329 阅读 · 0 评论 -
使用PyTorch中Dataset和Dataloader遇到的问题
今天在使用PyTorch中Dataset遇到了一个问题。先看代码class psDataset(Dataset): def __init__(self, x, y, transforms = None): super(Dataset, self).__init__() self.x = x self.y = y if tra...原创 2018-11-14 15:22:11 · 6625 阅读 · 0 评论 -
使用PyTorch将文件夹下的图片分为训练集和验证集
PyTorch提供了ImageFolder的类来加载文件结构如下的图片数据集:root/dog/xxx.pngroot/dog/xxy.pngroot/dog/xxz.pngroot/cat/123.pngroot/cat/nsdf3.pngroot/cat/asd932_.png使用这个类的问题在于无法将训练集(training dataset)和验证集(validation ...原创 2018-11-21 16:32:22 · 10525 阅读 · 5 评论