Pytorch
文章平均质量分 55
FL1623863129
计算机应用软件开发,长期致力于研究图像算法,人工智能算法,边缘部署,算法落地研究
展开
-
[pytorch][原创]获取torch tensor中指定条件的数据
代码场景:在目标检测中获取到了类别,置信度和bbox的tensor,但是我想只想要指定类别的数据集咋办,比如只要class=1的数据,于是便有了下面代码import torchcls = torch.Tensor([1, 1, 2, 3, 1]) conf = torch.Tensor([0.1, 0.2, 0.3, 0.4, 0.5])bbox = torch.Tensor([[1, 1, 1, 1], [2, 2, 2, 2], [3, 3, 3, 3], [4, 4, 4, 4], [5.原创 2022-05-08 19:18:46 · 1266 阅读 · 0 评论 -
[deepstream][原创]deepstream:5.1-21.02-triton的docker无法使用pytorch
deepstream:5.1-21.02-triton的docker安装pytorch后会提示libtorch_cuda_cpp.so: undefined symbol参照Unable to Import PyTorch - #4 by mchi - DeepStream SDK - NVIDIA Developer Forums解决方法it’s caused by PyTorch version incompatibilities.After installing torch, remov原创 2021-10-28 18:10:09 · 1654 阅读 · 0 评论 -
[pytorch][转载]官方历史版本pytorch安装指令合集
We’d prefer you install thelatest version, but old binaries and installation instructions are provided below for your convenience.COMMANDS FOR VERSIONS >= 1.0.0v1.8.0CondaOSX# condaconda install pytorch==1.8.0 torchvision==0.9.0 torchaudio=.原创 2021-05-05 09:02:54 · 687 阅读 · 0 评论 -
[pytorch][转载]PyTorch中使用指定的GPU
PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。有如下两种方法来指定需要使用的GPU。1. 类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。1.1 直接终端中设定:CUDA_VISIBLE_DEVICES=1 python my_script.py1.2 python代码中设定:import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"见网址:http://原创 2021-04-22 09:27:09 · 117 阅读 · 0 评论 -
[pytorch][原创]yolov5的80类别名
提取自yolov5 3.1版本的类别名personbicyclecarmotorcycleairplanebustraintruckboattraffic lightfire hydrantstop signparking meterbenchbirdcatdoghorsesheepcowelephantbearzebragiraffebackpackumbrellahandbagtiesuitcasefrisbeeskissnowbo.原创 2021-03-01 14:51:58 · 2020 阅读 · 1 评论 -
[pytorch][转载]torch.onnx.export():将pytorch模型转换为.onnx模型
torch.onnx.export(model, args, f, export_params=True, verbose=False, training=False, input_names=None, output_names=None, aten=False, export_raw_ir=False, operator_export_type=None, opset_version=None, _retain_param_name=True, do_constant_folding=False, e.原创 2020-11-22 14:20:46 · 3723 阅读 · 5 评论 -
[Pytorch][转载]Running Pytorch models in production
PytorchandTensorfloware two widely used frameworks that have become today’s standard when it comes to deep learning. Tensorflow is Google’s child, released in 2015, and has been the most famous deep learning framework ever since, but now there is a new ...原创 2020-11-21 16:01:18 · 250 阅读 · 0 评论 -
[pytorch][转载]pytorch-yolov3视频测试代码
项目地址:https://github.com/eriklindernoren/PyTorch-YOLOv3由于作者没有给出视频检测代码。下面测试通过自定义模型视频检测from __future__ import divisionfrom models import *from utils.utils import *from utils.datasets import *import osimport sysimport timeimport datetimeimport ar原创 2020-11-21 09:04:35 · 538 阅读 · 1 评论 -
[pytorch][原创]保存模型和权重
只保存模型参数# 保存torch.save(model.state_dict(), '\model.pth')# 加载model = TheModelClass(...)model.load_state_dict(torch.load('\model.pth'))保存完整模型# 保存torch.save(model, '\model.pth')# 加载model = torch.load('\model.pth')...原创 2020-10-28 13:48:56 · 1332 阅读 · 0 评论 -
[Pytorch][转载]resnet模型实现
本文源自Pytorch官方:https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.pyimport torchimport torch.nn as nnfrom .utils import load_state_dict_from_url__all__ = ['ResNet', 'resnet18', 'resnet34', 'resnet50', 'resnet101', ...原创 2020-05-28 15:09:51 · 409 阅读 · 0 评论 -
[Pytorch][转载]VGG模型实现
本文源自Pytoch官方:https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.pyimport torchimport torch.nn as nnfrom .utils import load_state_dict_from_url__all__ = [ 'VGG', 'vgg11', 'vgg11_bn', 'vgg13', 'vgg13_bn', 'vgg16', 'vgg16_...原创 2020-05-28 15:06:32 · 372 阅读 · 0 评论 -
[Pytorch][转载]AlexNet模型实现
本文源自Pytorch官方:https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.pyimport torchimport torch.nn as nnfrom .utils import load_state_dict_from_url__all__ = ['AlexNet', 'alexnet']model_urls = { 'alexnet': 'https:...原创 2020-05-28 14:58:40 · 408 阅读 · 0 评论 -
[Pytorch][转载]用pytorch实现两层神经网络
PyTorch: Tensors这次我们使用PyTorch tensors来创建前向神经网络,计算损失,以及反向传播。一个PyTorch Tensor很像一个numpy的ndarray。但是它和numpy ndarray最大的区别是,PyTorch Tensor可以在CPU或者GPU上运算。如果想要在GPU上运算,就需要把Tensor换成cuda类型。import torch...原创 2020-04-23 17:38:38 · 394 阅读 · 0 评论 -
[Pytorch][转载]用pytorch解决简单的FizzBuzz问题
FizzBuzz FizzBuzz是一个简单的小游戏。游戏规则如下:从1开始往上数数,当遇到3的倍数的时候,说fizz,当遇到5的倍数,说buzz,当遇到15的倍数,就说fizzbuzz,其他情况下则正常数数。 我们可以写一个简单的小程序来决定要返回正常数值还是fizz, buzz 或者 fizzbuzz。 # One-hot encode the de...原创 2020-04-23 17:37:34 · 458 阅读 · 0 评论 -
[Pytorch][转载]用numpy实现两层神经网络
一个全连接ReLU神经网络,一个隐藏层,没有bias。用来从x预测y,使用L2 Loss。 这一实现完全使用numpy来计算前向神经网络,loss,和反向传播。 numpy ndarray是一个普通的n维array。它不知道任何关于深度学习或者梯度(gradient)的知识,也不知道计算图(computation graph),只是一种用来计算数学运...原创 2020-04-21 22:56:04 · 501 阅读 · 0 评论