![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
大别山伧父
这个作者很懒,什么都没留下…
展开
-
Siamese目标跟踪系列学习笔记
1. 网络结构消除padding对跟踪过程的影响def center_crop(x): """ center crop layer. crop [1:-2] to eliminate padding influence. Crop 1 element around the tensor input x can be a Variable or Tensor """ return x[:, :, 1:-1, 1:-1].contiguous()带有pa原创 2021-06-17 19:58:01 · 324 阅读 · 0 评论 -
got-10k数据预处理
使用shutil移动got10k文件夹# -*- coding: utf-8 -*- import shutilimport os dst = r'C:\Users\lpf\Desktop\dataset\Got10k\train'dst1 = r'd:/Learning_Data/BasicCode/move_file/c'path = []def Test1(rootDir, path): list_dirs = os.walk(rootDir) for root, di原创 2021-06-15 19:55:13 · 2985 阅读 · 12 评论 -
pytorch 张量处理
张量处理张量基本信息tensor = torch.randn(3,4,5)print(tensor.type()) # 数据类型print(tensor.size()) # 张量大小print(tensor.dim()) # 维度的数量张量命名NCHW = [‘N’, ‘C’, ‘H’, ‘W’] images = torch.randn(32, 3, 56, 56, names=NCHW) images.sum('C') images.select('C', index=0)原创 2021-06-15 14:56:45 · 578 阅读 · 0 评论 -
pytorch和opencv读取图片的操作
opencv读取图片后,图片格式变为BGRpytorch处理的张量图片格式为RGB若使用cv2来读取图片,在使用pytorch模型来对图片数据进行后续运算,一定要将BGR转换为RGB,代码如下import cv2import numpy as npimg_o = cv2.imread(img_path) # BGRimg = img[:, :, ::-1].transpose(2, 0, 1) # BGR to RGB, to 3x416x416img = np.ascontiguo.原创 2021-06-08 15:38:29 · 1137 阅读 · 1 评论 -
torchvision.datasets.ImageFolder用法
torchvision之ImageFolder用法当前文件夹:/root-- train/---- 001.jpg---- 002.jpg---- ....-- test/---- 001.jpg---- 002.jpg---- ....导入当前文件夹的训练集,用下面的代码会报错root = "./root/train"train_dataset = ImageFolder(root = root_path)正确的路径data/├── train/| ├── cla原创 2021-06-05 16:51:40 · 1660 阅读 · 1 评论