

YOLO模型

论文给出了一个24层卷积的CNN模型

实现

• py/lib/models/basic_conv2d.py
• py/lib/models/yolo_v1.py

模型输入4维数据（大小为[N, C, H, W]）；输出3维数据（大小为[N, S*S, B*5+C])