参考链接:
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