![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
机器学习
文章平均质量分 75
lealaxy
这个作者很懒,什么都没留下…
展开
-
用docker构建能用GPU炼丹的CUDA容器,附PyToch版本Dockerfile
背景实验室的服务器内核版本太老了,有一个项目需要升级内核,但是服务器上还有一堆东西不敢随便升级。于是就准备用docker构建一个镜像,安装CUDA和Python环境,平时ssh连进去炼丹需求炼丹必备的cuda肯定是必不可少的,ssh服务器也需要配置,既然准备写一个dockerfile,那python环境和换源之类的也就一块打包到镜像里去得了。以后谁想炼丹直接新建一个容器,映射好端口之后容器里炼丹的基础设施就都有了。安装宿主机安装CUDA驱动想要容器能用CUDA,宿主机肯定要安装CUDA驱动,原创 2022-05-24 19:15:19 · 1972 阅读 · 2 评论 -
基于自定义数据集的Caffe模型生成
[图片上传失败…(image-e3657c-1587957212331)]首先,整个的工程目录大概是这样的:每个文件对应的功能会在后面的文章中介绍,现在可以先不急于创建整个工程目录。准备数据集给数据集重新命名每一张图片前都插入图片的标签,可以使用ubuntu的批量重命名的功能.比如说我要做一个二分类的网络,命名格式应该是这样的:其中每个图片的开头都是图片的标签,这样方便与生成图片txt...原创 2020-04-27 11:14:29 · 301 阅读 · 0 评论 -
YOLOV4首发使用体验
YOLOV4首发使用体验[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-p5i1YLfk-1587914617755)(https://pjreddie.com/media/image/yologo_2.png)]YOLOV4训练自己的数据集训练配置基本上就是官方文档的翻译,加上一些自己的理解,英语好的可以直接看官方文档更改网络配置参数(cfg文件)...原创 2020-04-26 23:24:27 · 1358 阅读 · 0 评论