![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
小新GSUNG0222
这个作者很懒,什么都没留下…
展开
-
[CenterNet_Lite]使用自己的数据集训练CenterNet_Lite,从安装到使用
1、安装CenterNet_Lite1.1 环境配置先给出源码git地址:https://github.com/yjh0410/CenterNet-Lite我的服务器的环境:ubuntu16.04 、CUDA Version 9.0.176、CUDNN 7.5.0我使用了py3.6,pytorch==1.2.0 与之匹配的torchvision==0.4.0首先创建了一个新的虚拟环境:conda create -n CenterNetLite python=3.6source.原创 2020-11-30 20:21:40 · 2595 阅读 · 12 评论 -
[YOLO V5]使用自己的数据集训练YOLO V5,从安装到使用
1. 安装yolo v51.1 clone代码到自己的文件夹下cd $INSTALL_DIRgit clone https://github.com/ultralytics/yolov5.githttps://github.com/ultralytics/yolov51.2 根据requirements.txt安装依赖博主自己的环境是ubuntu 16.04,Driver Version: 430.50 ,CUDA Version: 10.1原创 2020-08-05 23:56:57 · 2983 阅读 · 0 评论 -
[YOLO V5] 官方预训练权重
一.YOLO V5官方代码https://github.com/ultralytics/yolov5二.预训练权重文件博主用网盘分享出来,便于没有梯子的同学使用https://pan.baidu.com/s/1yytOWpcV8SwUSYWMAFELmAdv17内容包括:原创 2020-07-27 10:44:32 · 9167 阅读 · 3 评论 -
解决PyTorch报错:RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatT
1. 问题描述1.1 在使用pytorch0.4推理pyramidbox 或者SSD时,发生一下报错:在pytorch0.3上面应该时不报错的,改成0.4之后发生报错。Traceback (most recent call last): File "test.py", line 241, in <module> det0 = detect_face(image...原创 2020-05-06 23:18:32 · 1159 阅读 · 1 评论