人工智能
IT捕快
这个作者很懒,什么都没留下…
展开
-
XML to COCO数据
XML to COCO数据XML to COCO数据XML to yoloXML to COCO数据# coding:utf-8# pip install lxmlimport osimport globimport jsonimport shutilimport numpy as npimport xml.etree.ElementTree as ETpath2 = "/raid/ZXF/project_jishi/data/物料堆放/coco" # 输出文件夹classes原创 2021-08-09 15:50:20 · 330 阅读 · 0 评论 -
opencv+opencv_contrib编译
opencv+opencv_contrib编译opencv下载地址opencv_contrib下载地址opencv+opencv_contrib编译方法opencv下载地址opencv下载opencv_contrib下载地址opencv_contrib下载opencv+opencv_contrib编译方法cd ~/opencv #进入opencv 源码下,不是opencv_contribmkdir buildcd buildcmake -DOPENCV_EXTRA_MODULES_PA原创 2021-07-21 15:07:07 · 1692 阅读 · 4 评论 -
record-----torch(.pth)转libtorch(.pt)的方法
torch转libtorchcodecode"""进行模型转换,torch->onnx->tvm环境要求:tvm转torch模型必须在pytorch1.4的环境下进行,1.4下可以进行torch->onnx->tvm,同时也可以进行torch->libtorch转换,但是c++中源码编译的libtorch库要和torch->libtorch转换时使用的pytorch版本一致由于c++工程中使用的libtorch库是1.0,因此torch->libtorc原创 2021-04-13 19:44:18 · 1992 阅读 · 0 评论 -
record-----torch(.pth)模型转onnx(.onnx)模型的方法
torch转onnx前言code前言torch转onnx目的用于将torch(.pth)模型转为tensorRT(.engine),pth->onnx->enginecode"""需要的环境:pytorch1.4进行模型转换,torch->onnx环境要求:tvm转torch模型必须在pytorch1.4的环境下进行,1.4下可以进行torch->onnx->tvm,同时也可以进行torch->libtorch转换,但是c++中源码编译的libtorch原创 2021-04-13 19:51:46 · 1305 阅读 · 0 评论 -
record-----各种实验指标记录
指标记录集前言指标tensorRT实验记录前言本文章记录工作中所完成的各种模型、实验的指标记录,便于后期查询使用。指标tensorRT实验记录以下是实验记录了各个模型在tensorRT多batch加速下的处理速度...原创 2021-05-13 15:13:28 · 175 阅读 · 0 评论