| 196 | |
| 197 | |
| 198 | def build_ssd(phase, size=300, num_classes=21): |
| 199 | if phase != "test" and phase != "train": |
| 200 | print("ERROR: Phase: " + phase + " not recognized") |
| 201 | return |
| 202 | if size != 300: |
| 203 | print("ERROR: You specified size " + repr(size) + ". However, " + |
| 204 | "currently only SSD300 (size=300) is supported!") |
| 205 | return |
| 206 | base_, extras_, head_ = multibox(vgg(base[str(size)], 3), |
| 207 | add_extras(extras[str(size)], 1024), |
| 208 | mbox[str(size)], num_classes) |
| 209 | return SSD(phase, size, base_, extras_, head_, num_classes) |