33 lines
831 B
Python
33 lines
831 B
Python
import numpy as np
|
|
|
|
from vision.utils.box_utils import SSDSpec, SSDBoxSizes, generate_ssd_priors
|
|
|
|
image_size = 300
|
|
image_mean = np.array([127, 127, 127]) # RGB layout
|
|
image_std = 128.0
|
|
iou_threshold = 0.45
|
|
center_variance = 0.1
|
|
size_variance = 0.2
|
|
|
|
specs = [
|
|
SSDSpec(19, 16, SSDBoxSizes(60, 105), [2, 3]),
|
|
SSDSpec(10, 32, SSDBoxSizes(105, 150), [2, 3]),
|
|
SSDSpec(5, 64, SSDBoxSizes(150, 195), [2, 3]),
|
|
SSDSpec(3, 100, SSDBoxSizes(195, 240), [2, 3]),
|
|
SSDSpec(2, 150, SSDBoxSizes(240, 285), [2, 3]),
|
|
SSDSpec(1, 300, SSDBoxSizes(285, 330), [2, 3])
|
|
]
|
|
|
|
priors = generate_ssd_priors(specs, image_size)
|
|
|
|
# print(' ')
|
|
# print('SSD-Mobilenet-v1 priors:')
|
|
# print(priors.shape)
|
|
# print(priors)
|
|
# print(' ')
|
|
|
|
# import torch
|
|
# torch.save(priors, 'mb1-ssd-priors.pt')
|
|
|
|
# np.savetxt('mb1-ssd-priors.txt', priors.numpy())
|