一、YOLO-v4主要做了什么?
YOLO-v4算法是在原有YOLO目标检测架构的基础上,采用了近些年CNN领域中最优秀的优化策略,从数据处理、主干网络、网络训练、激活函数、损失函数等各个方面都有着不同程度的优化,虽没有理论上的创新,但是会受到许许多多的工程师的欢迎,各种优化算法的尝试。
工业级别的目标检测关注的不仅仅是精度,还有速度,能达到实时是最理想状态,一般来讲,目标检测实时大于12.5fps被认为是实时,下面用TX2来测试看能不能达到实时的效果。
二、在TX2上的实验
1、下载源码
git clone https://github.com/AlexeyAB/darknet.git yolo4
2、配置Makefile文件
此处的ARCH的配置可以在下面的注释中找到,这是针对TX2的配置,将之粘贴上去,然后将先前的注释掉就可以了。
NVCC的路径需要自己找,可以使用如下的命令: