配套CV图像AI库的火山深度神经网络训练工具

maoxiaosc · 发表于 2022-10-18 20:23:05

本帖最后由 maoxiaosc 于 2022-10-27 18:08 编辑

本工具支持Caffe SSD目标检测网络训练,Darknet的Yolov4-v7目标检测及resnet、densnet分类网络的训练,拥有完全脱离python环境的数据集处理功能。Caffe训练

Darknet训练

训练后得到的权重模型CV图像AI库可以直接使用.
下载链接:之前上传的版本SSD有问题，重新上传一份链接：https://pan.baidu.com/s/1quIHMTcCS7zqEk8g54Va8A?pwd=5erl
提取码：5erl

maoxiaosc · 发表于 2022-10-26 09:31:49

看下说明，要安装cuda11.4,cudann8.2.2.26

maoxiaosc · 发表于 2022-11-1 11:34:00

372607507 发表于 2022-10-31 23:32
大神~是不是一定要CUDA11.4才可以Darknet训练？

是的，要CUDA11.4，CUDNN8.2.2.26

		自动登录	找回密码
密码			立即注册