forked from mindspore-Ecosystem/mindspore
65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
# Copyright 2020 Huawei Technologies Co., Ltd
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ============================================================================
|
|
|
|
"""Config parameters for SSD models."""
|
|
|
|
|
|
class ConfigSSD:
|
|
"""
|
|
Config parameters for SSD.
|
|
|
|
Examples:
|
|
ConfigSSD().
|
|
"""
|
|
IMG_SHAPE = [300, 300]
|
|
NUM_SSD_BOXES = 1917
|
|
NEG_PRE_POSITIVE = 3
|
|
MATCH_THRESHOLD = 0.5
|
|
|
|
NUM_DEFAULT = [3, 6, 6, 6, 6, 6]
|
|
EXTRAS_IN_CHANNELS = [256, 576, 1280, 512, 256, 256]
|
|
EXTRAS_OUT_CHANNELS = [576, 1280, 512, 256, 256, 128]
|
|
EXTRAS_STRIDES = [1, 1, 2, 2, 2, 2]
|
|
EXTRAS_RATIO = [0.2, 0.2, 0.2, 0.25, 0.5, 0.25]
|
|
FEATURE_SIZE = [19, 10, 5, 3, 2, 1]
|
|
SCALES = [21, 45, 99, 153, 207, 261, 315]
|
|
ASPECT_RATIOS = [(1,), (2, 3), (2, 3), (2, 3), (2, 3), (2, 3)]
|
|
STEPS = (16, 32, 64, 100, 150, 300)
|
|
PRIOR_SCALING = (0.1, 0.2)
|
|
|
|
|
|
# `MINDRECORD_DIR` and `COCO_ROOT` are better to use absolute path.
|
|
MINDRECORD_DIR = "MindRecord_COCO"
|
|
COCO_ROOT = "coco2017"
|
|
TRAIN_DATA_TYPE = "train2017"
|
|
VAL_DATA_TYPE = "val2017"
|
|
INSTANCES_SET = "annotations/instances_{}.json"
|
|
COCO_CLASSES = ('background', 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus',
|
|
'train', 'truck', 'boat', 'traffic light', 'fire', 'hydrant',
|
|
'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog',
|
|
'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra',
|
|
'giraffe', 'backpack', 'umbrella', 'handbag', 'tie',
|
|
'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball',
|
|
'kite', 'baseball bat', 'baseball glove', 'skateboard',
|
|
'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup',
|
|
'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
|
|
'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza',
|
|
'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed',
|
|
'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote',
|
|
'keyboard', 'cell phone', 'microwave oven', 'toaster', 'sink',
|
|
'refrigerator', 'book', 'clock', 'vase', 'scissors',
|
|
'teddy bear', 'hair drier', 'toothbrush')
|
|
NUM_CLASSES = len(COCO_CLASSES)
|