- 博客(10)
- 收藏
- 关注
原创 使用 TensorRT 进行 Depth Anything V2 推理
随着深度学习模型在计算机视觉任务中的应用日益广泛,深度估计任务作为其中的重要一环,广泛应用于自动驾驶、机器人和增强现实等领域。是一种高效的深度估计模型,其推理速度和准确性在众多应用场景中表现出色。然而,为了在实际应用中更好地发挥其性能,我们可以通过将模型部署在TensorRT上,从而进一步加速推理过程。本文将介绍如何将模型转换为TensorRT格式,并在实际应用中进行推理。
2024-10-11 17:15:46
856
6
原创 如何在YOLOv8网络中添加自定义注意力机制
在目标检测任务中,加入注意力机制可以提升模型的检测效果。本文将介绍如何在YOLOv8模型中集成多种注意力机制,如SimAMMHSACBAM和EMA,以增强模型对图像特征的提取能力。我们将展示每个注意力机制的代码示例,并讨论如何将这些模块添加到YOLOv8网络中。为了在YOLOv8中集成自定义的注意力机制模块,我们需要将相应的注意力机制代码文件(如SimAM.pyMHSA.pyCBAM.pyEMA.pyECA.py)放入项目中的目录下,并在task.py文件中进行相应的导入和修改。
2024-09-23 14:52:43
4156
7
原创 YOLOv8图像分割:使用ONNX模型进行推理
通过以上步骤,我们展示了如何使用YOLOv8进行图像分割的完整流程,从环境配置到代码实现和结果展示。此过程适用于YOLOv8目标分割任意模型进行分割任务。希望这篇博客能够帮助您理解和实现基于YOLOv8的目标分割项目。如果有任何问题或需要进一步的帮助,请随时留言讨论。
2024-03-19 16:18:46
2828
12
原创 YOLOv8+SAHI小目标检测:使用ONNX模型进行推理
SAHI是一个功能强大的小目标检测库,特别适合处理高质量图像中的小目标检测任务。尽管这种方法可能会导致处理时间增加,但它为需要高精度小目标检测的应用场景提供了一种有效的解决方案。希望本文能帮助你了解SAHI的使用场景和配置方法,如果有任何问题,欢迎留言讨论。
2024-02-28 18:06:04
5087
32
原创 AnimeGANv2动漫风格迁移:使用ONNX模型进行推理
AnimeGANv2是AnimeGAN的升级版本,旨在通过深度学习技术,捕捉并模仿特定动漫导演的风格,如宫崎骏、新海诚和今敏。它通过分析这些导演电影中的视频帧,学习如何将普通图片转化为具有相似艺术风格的动漫图像。这一过程不仅涉及颜色和纹理的转换,还包括细节和轮廓的调整,以达到高度逼真的动漫效果。
2024-02-27 14:35:39
1803
2
原创 深度学习入门:使用PyTorch和VGG16实现猫狗分类
这篇博客提供了使用PyTorch和VGG16进行猫狗分类的完整流程,包括数据处理、模型构建、训练、验证和推理。希望本教程能够为初学者提供一个良好的起点,鼓励他们在深度学习和人工智能的领域中继续探索和学习。
2024-02-23 13:10:25
953
原创 Gemma模型:谷歌的小巨人,让高性能AI触手可及
谷歌最近发布了一系列开放式大型语言模型,名为。这是谷歌加强对开源人工智能支持的一部分,也标志着Gemma与生态系统的完美集成。让我们深入了解Gemma及其如何帮助开发者和研究者在各种硬件上高效部署AI模型。
2024-02-22 19:00:25
1987
原创 YOLOv8人体关键点检测(姿态估计):使用ONNX模型进行推理
通过以上步骤,我们展示了如何使用YOLOv8进行人体关键点检测的完整流程,从环境配置到代码实现和结果展示。此过程适用于YOLOv8人体关键点检测任意模型进行检测任务。希望这篇博客能够帮助您理解和实现基于YOLOv8的人体关键点检测项目。如果有任何问题或需要进一步的帮助,请随时留言讨论。
2024-02-22 14:29:31
5871
3
原创 YOLOv8目标检测:使用ONNX模型进行推理
通过以上步骤,我们展示了如何使用YOLOv8进行目标检测的完整流程,从环境配置到代码实现和结果展示。此过程适用于YOLOv8目标检测任意模型进行检测任务。希望这篇博客能够帮助您理解和实现基于YOLOv8的目标检测项目。如果有任何问题或需要进一步的帮助,请随时留言讨论。
2024-02-22 11:14:34
11820
25
原创 构建与部署英中/中英翻译API:使用FastAPI和Hugging Face 模型
本文介绍了如何部署和使用Hugging Face预训练模型进行英文到中文以及中文到英文的翻译服务。我们通过FastAPI搭建了服务器,并展示了如何从客户端进行调用。希望这篇博客对你有所帮助!英文到中文模型中文到英文模型如有更多问题或特定需求,可以留言讨论。
2024-02-21 16:40:29
998
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人