深度学习
文章平均质量分 55
苏三福
我终于接受了黑色!
展开
-
coco数据集提取instance 和bbox
将图片转为xml,例:COCO_train2017_000000196610.jpg-->COCO_train2017_000000196610.xml。os.makedirs(path) # 可以创建多级目录。#coco有80类,这里写要提取类的名字,以person为例。#使用COCO API用来初始化注释数据。#获取COCO数据集中的所有类别。#通过id,得到注释的信息。原创 2023-02-20 10:35:09 · 731 阅读 · 0 评论 -
【无标题】yolov5 結果分析
yolov5训练结果解析_XiaoGShou的博客-CSDN博客_yolov5训练结果分析转载 2022-05-25 16:36:25 · 675 阅读 · 0 评论 -
yolov5-6.0 使用torchsummary 打印模型信息
TorchSummary无法载入Dict数据类型解决方法 - kenvision - 博客园torchsummary是一个比较不错的评价网络数据结构的包,但是目前torchsummary只支持网络输入和输出为torch.Tensor类型的网络,在对一些较为复杂的网络模型中,载入的可能并不一定为tensor类型,也有可能是list或者dict类型的数据。在train.py中的适当位置加入以下代码:from mysummary import summarysummary(model, torch原创 2022-03-30 10:31:52 · 2706 阅读 · 4 评论 -
查看ONNX模型node等信息
如何查看onnx网络结构和参数? - 知乎转载 2022-03-25 16:13:37 · 1363 阅读 · 0 评论 -
【pytorch转onnx,两个input】
import torchmodel_path = './hitnet_sf_finalpass/version_40/checkpoints/epoch=6-step=44890.ckpt'left_in = (torch.randn(1, 3, 480, 640, device='cuda'), torch.randn(1, 3, 480, 640, device='cuda'))right_in = torch.randn(1, 3, 480, 640, device='cuda')ck.原创 2022-03-23 18:27:55 · 2876 阅读 · 0 评论 -
Yolov5 libtorch 训练自己的数据并用liborch 部署
环境:ubuntu18.01(训练平台) , windows / vs2017 部署平台 opencv3.4.7 (提前编译好的)cuda10.1pytorch1.6yolov5 项目:https://github.com/ultralytics/yolov5yolov5v2.0模型下载链接:https://github.com/ultralytics/yolov5/releases训练阶段:libtorch部署 win10 vs2017, opencv3.4...原创 2021-02-27 16:44:50 · 2045 阅读 · 2 评论 -
Jeston nano 2GB 使用记录-烧录
Nvidia 最近更新了jeston nano 系列,刚出了款入门级别的jeston nano 2GB ;入手的第一款单片机,有必要做个记忆。双十一前夕,京东上做活动买的,不到434rmb 入手。果然便宜的连电源线都没有。。。而后又买了:电源线 :亚博智能YahBoom Jeston nano 2GB 电源适配器 5V4A 充电器type-C网线 : 绿联 六类cat6 千兆网络 2米屏幕:三星 27寸 2k IPS FreeSync HDMI F27T00QQC 三面微边 (双1.原创 2020-11-16 23:33:31 · 1156 阅读 · 3 评论 -
ubuntu18.04 cuda-10.1 cudnn7.6.5 安装tensorrt 6.0.1.5
还是有些坑要填。。。首先安装适合自己电脑配置的cuda,我这里之前安装好了cuda-10.1版本,cuda-10.1的环境变量记得添加:$ gedit ~/.bashrc 打开,在文件最后添加:export PATH=/usr/local/cuda-10.1/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH记得 $ source ~/.bashrc 一下。查看是否安装好的cuda.原创 2020-10-27 16:24:18 · 1532 阅读 · 1 评论