Jee_King的博客

因为处理数据需求，要对tensor实现 cv2.warpAffine( )的功能原图： size = 360 × 480首先cv2.warpAffine( )，import cv2import numpy as npimg_path = r'C:\Users\优快云\Desktop\00000001.jpg'image = cv2.imread(img_path)mapping2 = np.array([[ 1.23704604, 0., -256], [ 0.,

转自: https://blog.youkuaiyun.com/github_36923418/article/details/84567849如何根据自己需求设定，可学习参数，并进行初始化。#比如cnn输出4个东西，你又不想concate到到一起，你想用权重加法，权重又不想自己设定，想让网络自己学#requires_grad=True这个很重要#设置前置网络及 可学习参数self.cnn=cnn_...

代码逻辑和图片来自论文：Dynamic Multi-scale Filters for Semantic Segmentationclass DCM(nn.Module): def __init__(self, in_channel=64, out_channel=64, filter_size=3): super(DCM, self).__init__() ...

版本信息：python 3.6pytorch 0.4.1tensorflow-gpu 1.4.0tensorboard 1.6.0 [tensorflow 与 tensorboard版本匹配]from tensorboard_logger import Loggerlogger = Logger(logdir="./tensorboard_logs", flush_secs=10)...

脑子不好使，记录方便查阅...optimizer = optim.SGD(para_optim, lr=0.001, momentum=0.9, weight_decay=0.0002)def adjust_learning_rate(optimizer, epoch): lr = optimizer.param_groups[0]['lr'] * 0.1 #学习率每个epoch...

pip国内的一些镜像阿里云 https://mirrors.aliyun.com/pypi/simple/中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/豆瓣(douban) http://pypi.douban.com/simple/清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/中国科学技术大学...

pip install cupy 失败，出现如下的错误信息：Collecting cupy Using cached https://files.pythonhosted.org/packages/4d/08/2b25443d31ea34cdd259a689d67bc20f19df0c079e1f517ca1bd5bd7e6e2/cupy-6.2.0.tar.gz ERROR: Co...

1.import utilsVGG = networks.VGG19('vgg19.pth', feature_mode=True)VGG.to(device)VGG.eval()print('---------- Networks initialized -------------')utils.print_network(VGG)print('------------------...

1. Tensor 与 numpy之间转换# tensor to numpya = torch.FloatTensor(3,3)a = a.numpy()# numpy to tensora = np.ones(5)a = torch.from_numpy(a)2. Variable 转 numpy，Tensor [pytorch0.4 之后不再使用variable]# to...

一. pytorch 指定GPU （转自http://www.cnblogs.com/darkknightzh/p/6836568.html）PyTorch默认使用从0开始的GPU，如果GPU0正在运行程序，需要指定其他GPU。有如下两种方法来指定需要使用的GPU。1.类似tensorflow指定GPU的方式，使用CUDA_VISIBLE_DEVICES。1.1 直接终端中设定：CUDA...

找了一些资料，对我这种菜鸟并不友好，把自己摸索的相对详细的过程记录一下。pytorch 加载全部模型比较简单，直接使用如下代码:net.load_state_dict(torch.load(pth_path))现在只想使用 上述net（假设叫net-a）的中间的一部分模型，步骤如下：根据net-a的网络模型代码（如下），新建一个副本网络模型net-b代码。 class Net...

iccd@iccd-937:~$ python>>> import torch>>> print(torch.__version__) (两个下划线)