資源簡介
SSD使用VGG-16-Atrous作為基礎網絡,其中黃色部分為在VGG-16基礎網絡上填加的特征提取層。SSD與yolo不同之處是除了在最終特征圖上做目標檢測之外,還在之前選取的5個特特征圖上進行預測。
SSD圖1為SSD網絡進行一次預測的示意圖,可以看出,檢測過程不僅在填加特征圖(conv8_2, conv9_2, conv_10_2, pool_11)上進行,為了保證網絡對小目標有很好檢測效果,檢測過程也在基礎網絡特征圖(conv4_3, conv_7)上進行。

代碼片段和文件信息
“““Keras?implementation?of?SSD.“““
import?keras.backend?as?K
from?keras.layers?import?Activation
from?keras.layers?import?AtrousConvolution2D
from?keras.layers?import?Convolution2D
from?keras.layers?import?Dense
from?keras.layers?import?Flatten
from?keras.layers?import?GlobalAveragePooling2D
from?keras.layers?import?Input
from?keras.layers?import?MaxPooling2D
from?keras.layers?import?merge
from?keras.layers?import?Reshape
from?keras.layers?import?ZeroPadding2D
from?keras.models?import?Model
from?ssd_layers?import?Normalize
from?ssd_layers?import?PriorBox
def?SSD300(input_shape?num_classes=21):
????“““SSD300?architecture.
????#?Arguments
????????input_shape:?Shape?of?the?input?image
????????????expected?to?be?either?(300?300?3)?or?(3?300?300)(not?tested).
????????num_classes:?Number?of?classes?including?background.
????#?References
????????https://arxiv.org/abs/1512.02325
????“““
????net?=?{}
????#?Block?1
????input_tensor?=?input_tensor?=?Input(shape=input_shape)
????img_size?=?(input_shape[1]?input_shape[0])
????net[‘input‘]?=?input_tensor
????net[‘conv1_1‘]?=?Convolution2D(64?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv1_1‘)(net[‘input‘])
????net[‘conv1_2‘]?=?Convolution2D(64?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv1_2‘)(net[‘conv1_1‘])
????net[‘pool1‘]?=?MaxPooling2D((2?2)?strides=(2?2)?border_mode=‘same‘
????????????????????????????????name=‘pool1‘)(net[‘conv1_2‘])
????#?Block?2
????net[‘conv2_1‘]?=?Convolution2D(128?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv2_1‘)(net[‘pool1‘])
????net[‘conv2_2‘]?=?Convolution2D(128?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv2_2‘)(net[‘conv2_1‘])
????net[‘pool2‘]?=?MaxPooling2D((2?2)?strides=(2?2)?border_mode=‘same‘
????????????????????????????????name=‘pool2‘)(net[‘conv2_2‘])
????#?Block?3
????net[‘conv3_1‘]?=?Convolution2D(256?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv3_1‘)(net[‘pool2‘])
????net[‘conv3_2‘]?=?Convolution2D(256?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv3_2‘)(net[‘conv3_1‘])
????net[‘conv3_3‘]?=?Convolution2D(256?3?3
???????????????????????????????????activation=‘relu‘
???????????????????????????????????border_mode=‘same‘
???????????????????????????????????name=‘conv3_3‘)(net[‘conv3_2‘])
????net[‘pool3‘]?=?MaxPooling2D((2?2)?strides=(2?2)?border_mode=‘same‘
????
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2017-03-07?07:04??ssd_keras-master\
?????文件????????1070??2017-03-07?07:04??ssd_keras-master\.gitignore
?????文件????????1069??2017-03-07?07:04??ssd_keras-master\LICENSE
?????目錄???????????0??2017-03-07?07:04??ssd_keras-master\PASCAL_VOC\
?????文件????????3279??2017-03-07?07:04??ssd_keras-master\PASCAL_VOC\get_data_from_xm
?????文件?????????910??2017-03-07?07:04??ssd_keras-master\README.md
?????文件?????2511277??2017-03-07?07:04??ssd_keras-master\SSD.ipynb
?????文件??????419988??2017-03-07?07:04??ssd_keras-master\SSD_training.ipynb
?????文件?????1113254??2017-03-07?07:04??ssd_keras-master\gt_pascal.pkl
?????目錄???????????0??2017-03-07?07:04??ssd_keras-master\pics\
?????文件???????57184??2017-03-07?07:04??ssd_keras-master\pics\boys.jpg
?????文件???????45826??2017-03-07?07:04??ssd_keras-master\pics\car_cat.jpg
?????文件???????87894??2017-03-07?07:04??ssd_keras-master\pics\car_cat2.jpg
?????文件??????140391??2017-03-07?07:04??ssd_keras-master\pics\cat.jpg
?????文件???????47638??2017-03-07?07:04??ssd_keras-master\pics\fish-bike.jpg
?????文件??????344513??2017-03-07?07:04??ssd_keras-master\prior_boxes_ssd300.pkl
?????文件???????14599??2017-03-07?07:04??ssd_keras-master\ssd.py
?????文件????????6719??2017-03-07?07:04??ssd_keras-master\ssd_la
?????文件????????5751??2017-03-07?07:04??ssd_keras-master\ssd_training.py
?????文件???????10555??2017-03-07?07:04??ssd_keras-master\ssd_utils.py
?????目錄???????????0??2017-03-07?07:04??ssd_keras-master\testing_utils\
?????文件????????7677??2017-03-07?07:04??ssd_keras-master\testing_utils\videotest.py
?????文件?????????826??2017-03-07?07:04??ssd_keras-master\testing_utils\videotest_example.py
- 上一篇:基于結構光的3D掃描與重建初步
- 下一篇:機械設計齒輪設計程序源代碼
評論
共有 條評論