tkDNN使用步骤

参考链接:
https://github.com/ceccocats/tkDNN
https://github.com/ceccocats/tkDNN/issues/87

1.用darknet将权重转化为layers和debug

git clone https://git.hipert.unimore.it/fgatti/darknet.git
cd darknet
make
mkdir layers debug
./darknet export <path-to-cfg-file> <path-to-weights> layers

2.将debug和layers放到tkDNN/build/yolo4下面

3.如果是自己的数据集需要去tkDNN/tests/darknet下面找yolo4.cpp修改对应的输出层信息、cfg、names类别

4.整完了以后应该在build文件夹下执行
cmake … -DDEBUG=True
make

5.运行一下就可以得到结果了
rm yolo4_fp32.rt
./test_yolo4

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值