pytorch
新生代农民工!
人工智障
展开
-
训练分类模型一般能够用到的小技巧
分类模型训练trick;1: 网络模型的选择(resnet、densenet、EfficientNet等)2:数据增强的方式(翻转、亮度对比度、随机裁剪、随意移除等)3:损失函数(样本类别数量不均衡加权、难易程度focs loss)4:标签平滑 labelsmooth5:学习率 (warm-up、余弦退火)6:dropout与正则化 (优化器中的weight_decay)7:梯度累加(变相增加bathchsize)8:网络冻结(迁移学习冻结部分网络权重)9...原创 2020-11-22 12:24:57 · 644 阅读 · 0 评论 -
2020-10-24
https://github.com/open-mmlab/mmeditinghttps://zhuanlan.zhihu.com/p/178867385图像处理工具箱:分别有:图像修复,图像超分辨率,抠图,图像生成.原创 2020-10-24 19:53:31 · 212 阅读 · 0 评论 -
mmcv报错:ModuleNotFoundError: No module named ‘mmcv._ext‘
前几天安装mmedit训练超分辨率的模型,中途变了一下mmcv版本,然后又安装回去.这下坏了,无法运行.报错ModuleNotFoundError: No module named 'mmcv._ext'.我不断更换mmcv的版本一直各种各样的错误缺这缺那的然后文件夹的层也不对解决方法:mmcv正常安装先卸载mmcv-full,然后重新安装一下...原创 2020-08-21 17:01:31 · 7817 阅读 · 4 评论 -
mmdetection2.x windows下 PermissionError:
mmcv造成的shutil.copyfile(),复制文件,在复制的时候目标 文件是不能存在的,但是打印发现复制之前.py就已经存在。先将.py删掉,然后再复制就能出结果了。运行结果原创 2020-06-16 15:25:46 · 334 阅读 · 0 评论 -
pytorch:data读取出错:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 14 and 10 in dimension 1 at /pytorch/aten/src/TH/generic/THTensor.cpp:711原创 2020-06-11 12:10:42 · 3292 阅读 · 0 评论 -
resnet50、FPN、Panet结构及代码
起初faster-r-cnn,只采用最后一层特作为rpn以及head部分的特征图,后来不断改进,有了FPN, 一般来再后来有了Panet,网络的层数越深它的语义信息越丰富。但是随着网络层数的加深,检测所需的位置信息就会越差,CNN分类网络只需要知道一张图像的种类即可所以很多时候网络越深效果越好,但是不是分类效果越好的网越适合检测。FPN如下图所示,它用了不同大小的特征图进行预测,图中:下方的特征图较大,对应的感受野较小可以用来检测小目标,上部分的特征图尺寸较小,但是感受野较大适合检测大目标。...原创 2020-06-06 19:53:10 · 8754 阅读 · 7 评论 -
CNN分类网络之:Lenet
pytorch 写的一个 lenet的分类网络,不是百分百还原哈,结构是一样的, 简单训练一下自己的数据集。数据集格式如下,data内存放 自己的数据,每个类别放到一个文件夹中,文件夹名称为类别标签如下图 1.网络搭建import torchimport torch.nn as nnclass Lenet(nn.Module): def __in...原创 2020-06-02 21:33:43 · 396 阅读 · 0 评论 -
Pytorch 及 numpy 常用函数_个人记录不断加入
pytorch 编程时 用到的函数慢慢记录原创 2020-06-04 18:19:40 · 589 阅读 · 0 评论