【目标检测】yolov2特征提取网络------Darknet19结构解析及tensorflow和pytorch实现

1. 网络结构

Darknet-19 是 YOLO v2 的 backbone。
Darknet-19 总共有 19 层 conv 层, 5 个 maxpooling 层。Darknet-19 吸收了 VGG16, NIN 等网络的优点,网络结构小巧,但是性能强悍。
【目标检测】yolov2特征提取网络------Darknet19结构解析及tensorflow和pytorch实现_第1张图片
可以看到, Darknet-19 的 stride 为32, 没有 fc 层,而是用了 Avgpool。

2. 性能

下图截取自 Darknet 官网。可以看到 Darknet-19 的性能基本与 Resnet34 差不多。
【目标检测】yolov2特征提取网络------Darknet19结构解析及tensorflow和pytorch实现_第2张图片

DarkNet19模型------tensorflow自定义实现

import tensorflow as tf
from tensorflow.keras.layers import Conv2D,

你可能感兴趣的:(目标检测,tensorflow,目标检测,网络)