Pytorch
石头儿啊
follow me
展开
-
Yolov5 4.0版 网络结构图
前段时间下载的yolov5代码,本人最近想要学习一下,发现网上博主江大白的网络结构图和我下载的程序不太匹配,因此模仿大佬的结构 重新画了一个yolov5s的4.0版本的网络结构图。如果图式内容有问题还请各位不吝指教。原图:yolov5s 4.0修改版:...原创 2021-12-30 15:44:24 · 1347 阅读 · 2 评论 -
torch 测试GPU能否正常使用
运行程序import torchflag = torch.cuda.is_available()print(flag)ngpu= 1# Decide which device we want to run ondevice = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")print(device)print(torch.cuda.get_device_name(0))pri原创 2021-03-09 13:16:32 · 5715 阅读 · 7 评论 -
(解决方案)使用随机擦除后报错AttributeError: ‘NoneType’object has no attribute ‘shape’
问题描述: 最近我想要在图像数据增强阶段加入随机擦除transforms.RandomErasing()加入后发现一直报错AttributeError: ‘NoneType’ object has no attribute ‘shape’,查了很多都是在说opencv巴拉巴拉的问题,没有与我类似的问题,因此把解决方案记录下来。其实问题很简单,我之前一直是把transforms.RandomErasing()加在随机水平翻转transforms.RandomHorizontalFlip()等方法前后,经原创 2020-11-13 16:33:06 · 733 阅读 · 1 评论 -
Pytorch-YOLOV4-火焰目标检测
首先感谢大佬提供的代码bubbliiiing0.效果展示 1.所需环境torch==1.2.02.注意事项代码中的yolo4_weights.pth是基于608x608的图片训练的,代码中的默认anchors是基于608x608的图片的。注意不要使用中文标签,文件夹中不要有空格!在训练前需要务必在model_data下新建一个txt文档,文档中输入需要分的类,在train.py中将classes_path指向该文件。3.小技巧的设置在train.py文件下:1、mosaic参原创 2020-08-06 16:08:49 · 5511 阅读 · 49 评论