使用libtorch实现sobel算子进行边缘检测，python和c++实现

吴天德少侠

已于 2024-03-08 16:51:57 修改

阅读量464

点赞数 9

文章标签： python c++ 开发语言

于 2024-01-31 11:16:13 首次发布

本文链接：https://blog.csdn.net/sdhdsf132452/article/details/135949182

版权

发现有人拿我的博客文章去卖钱专栏收录该内容

63 篇文章 20 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何利用libtorch库在Python和C++中实现Sobel算子进行图像边缘检测。分别展示了Python和C++的实现结果。

摘要由CSDN通过智能技术生成

文章目录

python
- python结果
cpp
- c++结果

python

import torch
import torch.nn as nn
from PIL import Image
import numpy as np
import cv2


class SobelEdgeDetector(nn.Module):

了解本专栏

订阅专栏解锁全文

超级会员免费看

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

吴天德少侠

关注关注

9
点赞
踩
7

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

订阅专栏

Python实现Sobel算子进行边缘检测

持续更新

06-14

935

边缘检测是数字图像处理中的一个基本问题，其目的是在数字图像中找到边缘位置。Sobel算子是一种经典的边缘检测算法之一，它可以在图像中准确地检测出边缘位置。最终，我们便可以在窗口中看到经过Sobel算子处理后的边缘检测结果了。下面我们就来介绍如何使用Python实现Sobel算子进行边缘检测。Python实现Sobel算子进行边缘检测。

YOLOv5-LibTorch:通过LibTorch C ++ API部署YOLOv5进行实时对象检测

05-08

YOLOv5 LibTorch 通过LibTorch C ++ API部署YOLOv5进行实时对象检测环境 Ubuntu 18.04 OpenCV 3.2.0 LibTorch 1.6.0 CMake 3.10.2 入门安装OpenCV。 sudo apt-get install libopencv-dev 安装LibTorch。 wget https://download.pytorch.org/libtorch/nightly/cpu/libtorch-shared-with-deps-latest.zip unzip libtorch-shared-with-deps-latest.zip 编辑“ CMakeLists.txt”以正确配置OpenCV和LibTorch。编译并运行。 cd build cmake .. make ./../bin/YOLOv5

参与评论您还未登录，请先登录后发表或查看评论

LibTorch基础tensor运算

qq_43596950的博客

05-05

1551

/ 需要使用小写的tensor方法，而不是构建Tensor类 torch :: Tensor tensor = torch :: tensor({/*$ ./test2 3 45 6 7*//*$ ./test123*//*$ ./testNdArray:tensor:1 2 34 5 6*/

libtorch知识总结

m0_51661400的博客

12-01

2045

libtorch知识总结一， libtorch api1. 创建张量2. 属性3. 运算4. 形状变换5. 选择和切片6. 其它函数二，深度学习相关函数三，编译相关文件和命令1. g++命令2. Makefile3. CMakeLists.txt3.1 基本3.2 打包库(1) CMakeLists.txt主文件(2) sub CMakeLists.txt文件一， libtorch api 1. 创建张量 // randn 正态随机分布张量 t1 = torch::randn({3, 3}); //ran

深入理解LibTorch：从安装到API详解

最新发布

Chauvin的博客

06-29

1244

本文全面介绍了LibTorch的安装方法和关键API的使用，旨在为深度学习领域的开发者提供一份详实的指南，助力他们更高效地利用LibTorch进行模型开发与部署。无论是初学者还是有经验的开发者，都可以从中获得有价值的信息和实用的技巧。

c++ 部署libtorch时常用操作API

alex1801

11-10

7617

使用pytorch可以很方便的训练了网络，但是在部署libtorch的时候，c++对tensor的操作接口API，资料不是很多，因此，进行整理收集，形成文档。目录 1、tensor创建 1.1、std::vector转tensor 1.2、torch::empty() 1.3、torch::ones() 1.4、torch::zeros() 1.5、torch::rand() 2、获取tensor属性 2.1、tensor.sizes()/numel() 2.2、获取tensor值

canny算子、sobel算子边缘检测python代码

06-17

在Python中，同样利用OpenCV库实现Sobel算子边缘检测： ```python import cv2 import numpy as np # 读取图像 image = cv2.imread('input.jpg', 0) # 读取灰度图像 # Sobel算子计算x和y方向的梯度 sobel_x = cv2....

matlab边缘检测和图像分割1 Sobel算子图像边缘检测.zip

05-26

本资料包主要关注的是使用Sobel算子进行边缘检测。 边缘检测是图像处理中的基础操作，其目的是识别出图像中的边界，这些边界通常代表了图像中的对象或特征。Sobel算子是一种常用的微分算子，用于估计图像的梯度强度...

Sobel_sobel算子边缘检测_sobel算子_边缘检测_sobel边缘检测_tipa79_

10-03

在"tipa79"项目中，提供的"Sobel"文件很可能包含了一个实现Sobel算子边缘检测的代码示例。这个代码可能会涵盖以上所述的所有步骤，通过编程语言（如Python和OpenCV库）实现对图像的边缘检测，并输出处理后的灰度图像...

libtorch (pytorch c++) 教程（二）

浩瀚之水的专栏

05-12

2138

本章讲述张量的常见操作，可以先初步了解，具体在设计损失函数等任务中可以用到。随时翻阅张量初始化 libtorch(pytorch c++)的大多数api和pytorch保持一致，因此，libtorch中张量的初始化也和pytorch中的类似。本文介绍四种深度图像编程需要的初始化方法。第一种，固定尺寸和值的初始化。 //常见固定值的初始化方式 auto b = torch::zeros({3,4}); b = torch::ones({3,4}); b= torch::eye(4); b = to

[pytorch] libtorch-C++的实现

头狼博客

11-06

1482

libtorch是pytorch的C++版本，可以将pytorch的代码尤其自定义算子，通过libtorch迅速实现为C++版本的自定义算子，从而快速的实现模型部署的验证工作；

libtorch的初步使用

mujiangyao的博客

01-07

888

由于毕业设计的设计，学习使用了一下libtorch的使用。主要参考链接https://zhuanlan.zhihu.com/p/92374964。在写好cmake文件以及基本的加载代码之后，调用模型对单张图片进行深度预测，并将结果保存下来。代码如下： #include <torch/script.h> #include <iostream> #include <string> #include <memory> #include <ope..

c++ 部署libtorch 时对Tensor块的常用操作API (例如vector转换为torch、squeeze、select、select_indxe、max等）

请叫我小皇帝的博客

04-29

9447

使用pytorch可以很方便的训练了网络，并且pytorch的官方网站中放出了很全的python对tensor的操作接口API，但是在部署libtorch的时候，c++对tensor的操作接口API，资料不是很多哦，因此，我收集了我部署libtorch的时候，操作tensor块常用的接口API，其实很多和python的接口和类似。文章目录1、vector 转为 tensor 块2、torch::...

libtorch 常用api函数示例（史上最全、最详细）

qq_27009517的博客

01-04

1485

pytorch/libtorch qq群： 1041467052 其实pytorch的函数libtorch都有，只是写法上有些出入。libtorch的官方文档链接class tensor 只是官方文档只是类似与函数申明，没有告诉干嘛的，只能通过函数名字猜了。比如我要一个一个函数和已知的一个torch::Tensor变量形状一样，只是填充指定的数值，我记得在哪里看到过的有个full开头的函数，然后我就搜素full，然后找到一个函数full_like好像是我需要的。（见0）目录 pytorch/lib

libtorch推理实现

hajungong007的博客

08-27

1984

已经完成了pytorch C++的分类、语义分割、目标检测等任务的推理实现，在使用过程中遇到opencv4.0以上版本与libtorch不兼容的问题，是官方libtorch默认使用旧版c++11 ABI编译导致的，需要重新源码编译libtorch。参考了https://oldpan.me/archives/pytorch-c-libtorch-inference，以后有时间再完善本博客。 ...

pytorch中的unsqueeze以及squeeze用法举例

Warmth_Dream的博客

10-14

660

unsqueeze：简单来说就是添加tensor的维度举例说明：代码 import torch x = torch.tensor([1, 2, 3])#dim=1,即（3） print('x: ', x) print('x.size: ', x.size()) x1 = torch.unsqueeze(x, 1)#x1变为（3,1）的矩阵 print('x1: ', x1) print('x1.size: ', x1.size()) x2 = torch.unsqueeze(x, 0)#x2变

深入浅出PyTorch（算子篇）

cedi9117的博客

06-25

7310

解读PyTorch的内部机制。