
人工智能框架pytorch
仙凡兄
这个作者很懒,什么都没留下…
展开
-
windows vs2019libtorch不能成功调用cuda
需要新加命令行/INCLUDE:?warp_size@cuda@at@@YAHXZ,目前版本 libtorch-win-shared-with-deps-1.6.0+cu101,亲测可用。原创 2021-12-24 13:35:00 · 1801 阅读 · 1 评论 -
win10安装dcnv2
亲自安装1 git download dcnv2网址:下载2 打开vs2153 cd 到dcn2下载的路径在窗口运行python setup build develop注意:torch 1.8的环境失败,本人试的是torch1.3原创 2021-08-23 18:07:06 · 1415 阅读 · 0 评论 -
关于pytorch使用tensorboard以及visdom
UnicodeDecodeError: ‘utf-8’ codec can’t decode bytes in position 1022-1023:在python.py开头加上如下:#!/usr/bin/env Pythoncoding=utf-8原创 2021-01-19 16:40:29 · 366 阅读 · 1 评论 -
yolov5训练(心得)
yolov5成功训练真的是一波三折1 coco数据的下载(两种128,用于小批量测试,可以用来测试程序是否正常,真正训练的时候还是需要coco数据集,自己自定义的数据可以根据coco128.yaml文件夹的形式来定义,coco.yaml是txt的方式)2 运行程序1)window成功,虚拟机失败,服务器失败表象:如下图所示卡在那里不对排查原因:1:发现dataloader数据不对,循环打印,打印不出来,终于找到原因2:对比为什么windows可以,服务器不可以,发现windows安装的是t原创 2020-12-05 18:33:58 · 5282 阅读 · 7 评论 -
pytorch系列之实用数据增强
pytorch数据增强from PIL import Image, ImageFilterfrom torchvision import datasets,transforms img = Image.open(fn) #fn是图片路径 记住pytorch数据增强传入的格式是PIL !!!!mg = img.filter(ImageFilter.GaussianBlur(radius=1.1)) #模糊半径越大,图片就越模糊 高斯滤波 img = transforms.RandomRotatio原创 2020-09-28 10:59:33 · 648 阅读 · 0 评论