- 博客(16)
- 收藏
- 关注
原创 手撕代码:deep image matting (5)train和valid函数
vali和train函数的代码的一知半解def train(train_loader, model, optimizer, epoch, logger): model.train() # train mode (dropout and batchnorm is used) losses = AverageMeter() # Batches for i, (img, alpha_label) in enumerate(train_loa
2022-02-20 10:22:31
1985
原创 手撕代码1:deep image matting (4) DIM模型结构
dim的模型结构使用了Ctrl+F12生成出来的函数内部结构来看,变量这块实际上是缺失了,由于整体的代码长度太大,先看函数构造在细扣里面细节。init和forward是大部分继承pytorch的nn.Module模块的函数必须要具备的两个函数,init作为初始化函数,forward作为前向传播的函数。这里面有一个细节上的问题,在实际使用的时候并没有出现model.forwad()的代码,原因是直接使用函数模块调用DIMMod
2022-01-12 21:41:32
2705
2
原创 手撕代码1:deep image matting (2):ArgumentParse
对args对象进行解构这篇复现代码最舒服的地方就是把什么零件都在外面给你展示好了,至少不用在各种奇奇怪怪的文件夹里面翻箱倒柜。这里面进行解析的时候先把预处理的放到一边,从训练开始进行一点一点的解构,也就是train.pytrain.py 模型训练过程按照我的个人经验来说,从main函数进行拆解的话效率会特别高。那就直接进行拆解。简单粗暴到有点过分的程度,直接调用函数进行看似复杂的训练过程。首先声明.
2022-01-04 19:56:19
764
原创 手撕代码1:deep image matting
前言本系列文章就是为了解决一个特别坑的问题:下载下来的github模型该怎么去看懂。毕竟大多数从论文学代码的家伙的基础都是。。。差不多的拉。。。所以就把这个笔记作为一个记录弄上来了。代码github地址:https://github.com/foamliu/Deep-Image-Matting-PyTorch论文地址:https://arxiv.org/abs/1703.03872这篇论文被称为深度学习抠图的开山之作,在之后的深度学习抠图...
2022-01-04 10:35:07
3124
5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人