- 博客(3)
- 收藏
- 关注
原创 pytorch框架下的Finetune 以及ResNet50 代码
Resnet 50 # -*- coding: utf-8 -*- import torch.nn as nn import math import torch.utils.model_zoo as model_zoo class residual_block(nn.Module): expansion = 4 def __init__(self, inplanes...
2018-12-19 16:23:53
6464
3
原创 基于Resnet50的pytorch框架下的图像特征提取
在Resnet50CNN结构下实现图像的特征提取,这里采用的是CV2的图像读入方式,最后再把得到的图像转换成npy格式进行输出得得,图像对应的特征。 # -*- coding: utf-8 -*- """ Function: 图像特征的提取,可以依据需求修改CNN的输出,得到不同层网络的输出图像特征 Writer: Zenght date:2019.2.16 """ from __futur...
2019-02-16 16:32:32
16059
7
原创 利用Opencv模式进行图像读入下的Pytorch Finetune
# -*- coding: utf-8 -*- """ Auther : Haitao Zeng date : 2019.2.14 Function: finetune the pre-trained model on MIT67 & SUN397 """ from __future__ import print_function, division, absolute_import ...
2019-02-15 15:55:47
2338
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅