-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSSDNet.py
30 lines (25 loc) · 1 KB
/
SSDNet.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import math
from collections import namedtuple
import tensorflow as tf
import Config
slim = tf.contrib.slim
SSDParams = namedtuple('SSDParameters', ['img_shape',
'num_classes',
'no_annotation_label',
'feat_layers',
'feat_shapes',
'anchor_size_bounds',
'anchor_sizes',
'anchor_ratios',
'anchor_steps',
'anchor_offset',
'normalizations',
'prior_scaling'
])
class SSDNet(object):
def __init__(self):
return
default_params = SSDParams(
)
def BaseNet(self, input):
return