Ultra_net:针对DAC-SDC 2020的基于FPGA的对象检测
这是基于FPGA的神经网络推理的存储库。 该设计在获得了第一名。
BJUT_北京工业大学跑步者集团
詹康,郭俊南,宋冰岩,张文波*,包振山*
最终排名发布在
储存库组织
培训:包含培训脚本。
模型:包含预训练的权重,模型脚本和测试脚本。
量化:包含python脚本,该脚本处理要在Vivado HLS中使用的模型生成头文件。
hls:包含Ultra_net的Vivado HLS实现。
vivado:包含Vivado块设计文件。
部署:一个Jupyter笔记本,展示了如何使用基于FPGA的神经网络在ultra96-PYNQ上执行对象检测。
火车
cd火车/ yolov3 /
python3 train.py-多尺度--img大小320-多尺度-批处理大小32
量化
cd量化/
python3 torch
2022-12-13 20:48:51
6.01MB
C
1