pytorch cat、stack、tranpose、permute、unsqeeze

总结：torch版cuda最快，cpu次之，python最慢。

import time
import torch
import numpy as np
import cv2
if __name__ == '__main__':path=r'D:\111.jpg'img_ = cv2.imread(path)start = time.time()for j in range(500):#python中如果是除法，img=img/255.0 会很快;img1=img/255.0 就会比较慢，一张图片需要156msimg=img_/255.0image = np.transpose(img, (2, 0, 1))img2 = torch.from_numpy(image).float()#.unsqueeze(0)# time.sleep(0.007)print('time1',time.time()-start)img = cv2.imread(path)start = time.time()for j in range(500):# img_=img_/255.0img2 = torch.from_numpy(img)img2=img2.cuda().float().div(255.0)#.unsqueeze(0)img2=img2.permute(2, 0, 1)# time.sleep(0.007)print('time2', time.time() - start)

pytorch cat、stack、tranpose、permute、unsqeeze相关推荐

PyTorch学习笔记2：nn.Module、优化器、模型的保存和加载、TensorBoard
文章目录一.nn.Module 1.1 nn.Module的调用 1.2 线性回归的实现二.损失函数三.优化器 3.1.1 SGD优化器 3.1.2 Adagrad优化器 3.2 分层学习率 3 ...
PyTorch 1.7发布，支持CUDA 11、Windows分布式训练
机器之心报道参与:魔王.小舟昨日,PyTorch 团队发布 PyTorch 1.7 版本.该版本增添了很多新特性,如支持 CUDA 11.Windows 分布式训练.增加了支持快速傅里叶变换(FF ...
太棒了！PyTorch 1.7发布，支持CUDA 11、Windows分布式训练
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:机器之心 AI博士笔记系列推荐周志华<机器学习> ...
PyTorch 1.7 发布：支持 CUDA 11、FFT 新 API、及 Windows 分布式训练
PyTorch 1.7 现已发布,此版本增加了许多新的 API.包括对与 NumPy 兼容的 FFT 操作的支持.用于性能分析的工具.以及对基于分布式数据并行(Distributed Data Par ...
Pytorch：NLP 迁移学习、NLP中的标准数据集、NLP中的常用预训练模型、加载和使用预训练模型、huggingface的transfomers微调脚本文件
日萌社人工智能AI:Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战(不定时更新) run_glue.py微调脚本代码 python命令执行run ...
DL框架：主流深度学习框架(TensorFlow/Pytorch/Caffe/Keras/CNTK/MXNet/Theano/PaddlePaddle)简介、多个方向比较、案例应用之详细攻略
DL框架:主流深度学习框架(TensorFlow/Pytorch/Caffe/Keras/CNTK/MXNet/Theano/PaddlePaddle)简介.多个方向比较.案例应用之详细攻略目录深 ...
C++中的deque、stack、queue及priority_queue
C++中的deque.stack.queue及priority_queue 文章目录 C++中的deque.stack.queue及priority_queue 一.deque 二.stack 三.q ...
PyTorch框架学习十——基础网络层（卷积、转置卷积、池化、反池化、线性、激活函数）
PyTorch框架学习十--基础网络层(卷积.转置卷积.池化.反池化.线性.激活函数) 一.卷积层二.转置卷积层三.池化层 1.最大池化nn.MaxPool2d 2.平均池化nn.AvgPool2 ...
java stack 类效率_Java中ArrayList、LinkedList、Vector、Stack的比较
一.介绍先回顾一下List的框架图由图中的继承关系,可以知道,ArrayList.LinkedList.Vector.Stack都是List的四个实现类. AbstractList是一个抽象类,它 ...
【Java 集合】ArrayList、LinkedList、Stack、Queue、Set、Map，迭代器 Iterable、Iterator，Collections类
集合类型数组的局限性 ArrayList(常用方法 + 基本使用) ArrayList - retainAll ArrayList - toArray ArrayList 的遍历(5种) Array ...

pytorch cat、stack、tranpose、permute、unsqeeze

pytorch cat、stack、tranpose、permute、unsqeeze相关推荐

最新文章

热门文章