- 博客(86)
- 问答 (1)
- 收藏
- 关注
RSS订阅原创 python 多个文件夹下的图片 复制到 同意文件夹内，并重命名
一、代码import osimport shutilsrc = 'E:\\code\\TecoGAN-PyTorch-master\\data\\REDS_trans\\bicubic4xlr'target = 'E:\data\SISI_dataset\lr'path = os.listdir(src) # 000 001 002path.sort()count = len(path)# print(path)#['000', '001', '002',……‘029’]num =
2022-05-25 22:43:52
885
1
原创 numpy.core._exceptions.MemoryError: Unable to allocate 623. MiB for an array with shape (105, 3, 108
memoryerror记录一下，还不知道是否有效
2022-05-25 20:15:52
8230
1
原创 把进程挂在后台执行nohup
nohupnohup 英文全称 no hang up（不挂起），用于在系统后台不挂断地运行命令，退出终端不会影响程序的运行。执行命令：nohup /root/run.sh &&：让命令在后台执行，终端退出后命令仍旧执行。/root/run.sh：执行命令...
2022-04-22 16:15:23
499
原创 opencv视频逐帧转换为图片
#coding=utf-8import cv2import osimport os.path as ospname='big_road' #唯一变量video_path='video/'+name+'.mp4'pic_path='pic/'+nameif not osp.exists(pic_path): os.makedirs(pic_path)vc=cv2.VideoCapture(video_path)#读取视频if vc.isOpened(): #第一
2022-03-22 14:11:35
5603
1
原创 点乘与逐元素相乘
import torcha=torch.Tensor([[1,2,3,4,5],[6,7,8,9,10]])b=torch.Tensor([[2,4,6,8,10],[1,3,5,7,9]])逐元素相乘 element-wise multiplication#element-wise multiplicationc=a*bprint('c.size:\t',c.size(),'\nc:\t',c)输出：c.size: torch.Size([2, 5]) c: tensor([[
2022-03-11 15:26:43
3222
转载 ConvTranspose2d
FCN中一个很重要的部分—反卷积图片通过卷积层降低分辨率，提取特征，而反卷积则是把图片重新放大的一个结构。在语义分割中，必须对反卷积的反卷积核进行参数初始化(这点很重要)。一般使用的方法是双线性插值法。pytorch 中反卷积函数的说明：FCN中一个很重要的部分—反卷积图片通过卷积层降低分辨率，提取特征，而反卷积则是把图片重新放大的一个结构。在语义分割中，必须对反卷积的反卷积核进行参数初始化(这点很重要)。一般使用的方法是双线性插值法。pytorch 中反卷积函数的说明:给出反卷积操作输入尺
2022-01-13 16:59:16
214
转载 torch.nn.ReflectionPad2d()的用法简介
转载：https://blog.csdn.net/LionZYT/article/details/120181586最近在看Pix2PixHD的源码时，发现里面有几处用了nn.ReflectionPad2d()这个函数，查阅了官方文档和网上的资料后，终于弄明白了这个函数的用处。函数用途：对输入图像以最外围像素为对称轴，做四周的轴对称镜像填充。填充顺序：左->右->上->下对于一个4维的Tensor，当只指定一个padding参数时，则表示对四周采用相同的填充行数。# 对四周都填充
2022-01-10 14:04:10
658
原创 BasicVSR的模型net代码解析
完整代码如下：# Copyright (c) OpenMMLab. All rights reserved.import torchimport torch.nn as nnimport torch.nn.functional as Ffrom mmcv.cnn import ConvModulefrom mmcv.runner import load_checkpointfrom mmedit.models.common import (PixelShufflePack, Residua.
2022-01-06 13:32:19
4
原创 np.flip(img,axis=2)
import numpy as npfrom PIL import Imageimg=Image.open('222.png')img1=np.flip(img, axis=2)#翻转rgb中的r和bimg2=Image.fromarray(img1)img2.show()源222.png结果：
2021-12-28 17:43:45
2091
1
原创 realbasicvsr代码解析part1
RealBasicVSR的Inference_realbasicvsr.pymain():代码解析：def main(): args = parse_args() # initialize the model model = init_model(args.config, args.checkpoint) # read images file_extension = os.path.splitext(args.input_dir)[1]
2021-12-28 17:38:35
1450
原创 CUDA out of memory
RuntimeError: CUDA out of memory. Tried to allocate 4.35 GiB (GPU 0; 7.80 GiB total capacity; 4.37 GiB already allocated; 1.93 GiB free; 4.63 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to
2021-12-28 14:00:30
7520
空空如也
如何在代码层面区分数据并行和模型并行
2021-09-23
TA创建的收藏夹 TA关注的收藏夹
TA关注的人