No module named 'torch_scatter'

No module named 'torch_scatter' pip install --verbose --no-cache-dir torch-scatter $ pip install --verbose --no-cache-dir torch-sparse $ pip inst...

2019-04-08 00:14:08

阅读数 45

评论数 1

torchgeometry 安装

pip install torchgeometry

2019-04-07 23:55:13

阅读数 16

评论数 0

libtorchWindows中的使用

原文:https://oldpan.me/archives/pytorch-windows-libtorch Pytorch的C++端(libtorch)在Windows中的使用 Oldpan 2019年1月17日 36条评论1,929次阅读2人点赞 前言 填一个之前的坑啊,本篇的姊妹...

2019-03-21 11:03:18

阅读数 93

评论数 0

c++ opencv2 libtorch 读取预训练权重并进行预测 linux

c++ opencv2 libtorch 读取预训练权重并进行预测 原文:https://oldpan.me/archives/pytorch-c-libtorch-inference 本篇使用的平台为Ubuntu,Windows平台的请看Pytorch的C++端(libtorch)在Wind...

2019-03-21 11:00:14

阅读数 33

评论数 0

linux Pytorch C++ 环境搭建

Pytorch C++ 环境搭建 0,阅读Pytorch C++官网文档,在虚拟机上安装Ubuntu16.04的Linux操作系统搭建环境,并使用makefile进行编译,代码简洁易读。 sudo apt-get update sudo apt-get install make c...

2019-03-21 10:47:27

阅读数 21

评论数 0

windows10用c++部署libtorch过程中遇到的一些问题

windows10用c++部署libtorch过程中遇到的一些问题 libtorch1.0 vs2017 CMake3.14 windows10 无cuda 用c++调用pytorch模型官网上面有详细教程,也有很多博客,可以参考以下链接:https://blog.csdn.net/do...

2019-03-21 10:40:47

阅读数 72

评论数 0

Pytorch转TensorRT范例代码

原文:https://blog.csdn.net/quantum7/article/details/83380935 Pytorch转TensorRT范例代码   TensorRT官方文档说,/usr/src/tensorrt/samples/python/network_api_pytorch...

2019-03-20 11:15:30

阅读数 40

评论数 0

cuda runtime error (63) : OS call failed or operation not supported on this OS

cuda runtime error (63) : OS call failed or operation not supported on this OS Traceback (most recent call last): File "D:\ProgramData\Minic...

2019-03-18 21:28:49

阅读数 81

评论数 0

python 批量resize性能比较

torch是最快的,能快1ms 分别是5ms 9ms 6ms,但是torch 占用cpu很高。 import time import cv2 import torch img = cv2.imread('d:/guo.jpg') # modify the image path to...

2019-03-04 23:40:11

阅读数 32

评论数 0

only one element tensors can be converted to Python scalars

  only one element tensors can be converted to Python scalars       for neg_step in range(neg_,float(con_pos>=0.6).sum()//neg_+1,neg_):...

2019-02-13 00:34:43

阅读数 82

评论数 0

pytorch筛选修改问题

  pytorch筛选后修改,是不可以的: 自己总结的:一级筛选修改是ok的 筛选后,再进行筛选,修改二次筛选结果,原数据是不变的。 import torch a=torch.Tensor([0.5,0.6]) boxes=torch.Tensor([[0.1,0.2,0.62,0....

2019-02-02 10:18:28

阅读数 48

评论数 0

torch 排序

  按box 的x 排序 import torch idx=torch.Tensor(([0.1,0.2,0.5,0.6],[0.05,0.3,0.7,0.8])) a, rank = idx[:,0].sort(0,descending=False) #第2个维度排序 # the ra...

2019-02-01 15:47:55

阅读数 82

评论数 0

faceboxes显示负样本

  if int(con_neg[con_neg >= 0.8].size(0)) > 0: test_neg = torch.sigmoid(conf_preds[:, :, 0]) test_neg[pos_mask]=0 fo...

2019-02-01 14:57:09

阅读数 106

评论数 0

{TypeError}clamp(): argument 'min' must be Number, not Tensor

  {TypeError}clamp(): argument 'min' must be Number, not Tensor   x1是二维数组, xx1 = x1[order[1:]].clamp(min=x1[i]) 解决方法: x1=x1.cpu().data  

2019-02-01 12:32:19

阅读数 144

评论数 1

Layout of the output array img is incompatible with cv::Mat (step[ndims-1] !

Layout of the output array img is incompatible with cv::Mat (step[ndims-1]    https://blog.csdn.net/tqcai666/article/details/81751207   解决方法:调用co...

2019-01-30 20:20:41

阅读数 74

评论数 0

python、PyTorch图像读取与numpy转换

原文:https://blog.csdn.net/yskyskyer123/article/details/80707038 python、PyTorch图像读取与numpy转换 Tensor转为numpy np.array(Tensor) numpy转换为Tensor torch.Te...

2019-01-30 20:11:26

阅读数 102

评论数 0

invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1

  invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 原因:dataloader 需要的数据维度是一样的,有的有两个目标,有的有一个目标,就会出现异常: 解决方法:凑成一样的。 Trac...

2019-01-28 11:24:23

阅读数 554

评论数 0

pytorch多维筛选

    import torch x = torch.linspace(1, 8, steps=8).view(4, 2) #筛选第一维和第二维都>5.5的 print(x) area=(x[:,0]>5.5)&(x[:...

2019-01-24 11:31:08

阅读数 77

评论数 0

torch.topk

  import torch input=torch.Tensor([0.1,0.2]).cuda() k=3 v,k=torch.topk(input, k, dim=0, largest=True, sorted=True, out=None) print(v,k)

2019-01-23 19:50:18

阅读数 200

评论数 0

pytorch筛选统计

  大于统计: iou是可以的 n_correct= int(sum(iou > 0.5))   但是这个是不对的: aa= torch.sigmoid(conf_preds[:, :, 0][pos_mask]) n_correct= int(sum(aa&a...

2019-01-23 00:01:04

阅读数 75

评论数 0

提示
确定要删除当前文章?
取消 删除
关闭
关闭