![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
人工智能框架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 · 1643 阅读 · 1 评论 -
win10安装dcnv2
亲自安装 1 git download dcnv2 网址:下载 2 打开vs215 3 cd 到dcn2下载的路径 在窗口运行python setup build develop 注意: torch 1.8的环境失败,本人试的是torch1.3原创 2021-08-23 18:07:06 · 1292 阅读 · 0 评论 -
关于pytorch使用tensorboard以及visdom
UnicodeDecodeError: ‘utf-8’ codec can’t decode bytes in position 1022-1023: 在python.py开头加上如下: #!/usr/bin/env Python coding=utf-8原创 2021-01-19 16:40:29 · 295 阅读 · 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 · 5144 阅读 · 7 评论 -
pytorch系列之实用数据增强
pytorch数据增强 from PIL import Image, ImageFilter from 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 · 558 阅读 · 0 评论