![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 94
编程爱撞墙的铁头娃
偶尔做一点记录
展开
-
pytorch使用detectron2模型库模型训练自己的数据
一 应用场景在x86 (Ubuntu18.04)cpu,在pytorch1.10框架下,使用detectron2模型库模型训练自己的数据集,并进行目标检测推理。二 环境配置我的环境是:pytorch==1.10+cputorchvision==0.11.2+cpudetectron2==0.6opencv==4.5.51 基础环境配置这里可使用conda,或者python3-venv创建虚拟环境,这里不做赘述,仅讲解基础环境搭建。$ sudo apt-get i原创 2022-03-04 12:41:47 · 5402 阅读 · 5 评论 -
使用pytorch实现的yolov4训练自己的数据集,并进行推理
参考:https://github.com/Tianxiaomo/pytorch-YOLOv4/对该仓库的步骤详细描述了一下,并解决了部分问题,因为pytorch实现一 应用场景在x86 ,ubuntu18.04(cpu)上,使用pytorch实现的yolov4训练自己的数据集,并进行推理。二 环境准备该样例依赖以下环境:numpy==1.18.2tensorboardX==2.0scikit_image==0.16.2matplotlib==2.2.3tqdm=..原创 2022-03-03 12:30:56 · 4503 阅读 · 0 评论