Python generate_ssd_priorsの例

コード例 #1

ファイル: mobilenetv1_ssd_config.py プロジェクト: Deeplite/deeplite-torch-zoo

    def __init__(self):
        self.image_size = 300
        self.image_mean = np.array([127, 127, 127])  # RGB layout
        self.image_std = 128.0
        self.iou_threshold = 0.45
        self.center_variance = 0.1
        self.size_variance = 0.2

        self.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]),
        ]

        self.priors = generate_ssd_priors(self.specs, self.image_size)

コード例 #2

ファイル: vgg_ssd_config.py プロジェクト: Deeplite/deeplite-torch-zoo

    def __init__(self):
        self.image_size = 300
        self.image_mean = np.array([123, 117, 104])  # RGB layout
        self.image_std = 1.0

        self.iou_threshold = 0.45
        self.center_variance = 0.1
        self.size_variance = 0.2

        self.specs = [
            SSDSpec(38, 8, SSDBoxSizes(30, 60), [2]),
            SSDSpec(19, 16, SSDBoxSizes(60, 111), [2, 3]),
            SSDSpec(10, 32, SSDBoxSizes(111, 162), [2, 3]),
            SSDSpec(5, 64, SSDBoxSizes(162, 213), [2, 3]),
            SSDSpec(3, 100, SSDBoxSizes(213, 264), [2]),
            SSDSpec(1, 300, SSDBoxSizes(264, 315), [2])
        ]

        self.priors = generate_ssd_priors(self.specs, self.image_size)

コード例 #3

ファイル: moblienet_512_ssd_config.py プロジェクト: johsnows/ssd-migo-mobilentv3

import numpy as np

from vision.utils.box_utils import SSDSpec, SSDBoxSizes, generate_ssd_priors

image_size = 512
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])
# ]

specs = [
    SSDSpec(32, 16, SSDBoxSizes(60, 105), [2, 3]),
    SSDSpec(16, 32, SSDBoxSizes(105, 150), [2, 3]),
    SSDSpec(8, 64, SSDBoxSizes(150, 195), [2, 3]),
    SSDSpec(4, 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)