pytorch
pytorch
_liuzt
这个作者很懒,什么都没留下…
展开
-
NMS和几种IOU的复现
NMS和几种IOU的复现输入为一张图片以及图片经过Yolo v3的推理结果,下载链接,提取码ha4o。NMS的实现参考的yolo v3,在细节上做了一些优化实现了iou,giou,diou,ciou的计算import cv2import numpy as npimport randomimport torchimport mathdef bbox_giou(box1, box2): """ Returns the GIoU of two bounding boxe原创 2022-03-16 20:19:28 · 1093 阅读 · 0 评论 -
安装libtorch记录
安装libtorch记录环境win10 + vs2017 + pytorch1.6.0 + cmake3.12.1 + opencv3.4.5 + cuda10.2import torch as t print(t.version.cuda)print(t.__version__)10.21.6.0步骤下载对应版本的libtorch创建工程文件夹,包含build文件夹、example-app.cpp、CMakeLists.txt,使用cmake创建VS工程,需要设置对op原创 2020-11-06 18:01:31 · 811 阅读 · 0 评论