60 lines
1.3 KiB
Python
60 lines
1.3 KiB
Python
# ******************************************************************************
|
|
# Copyright (c) 2021-2022. Kneron Inc. All rights reserved. *
|
|
# ******************************************************************************
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class ImageType(Enum):
|
|
GENERAL = 'general'
|
|
BINARY = 'binary'
|
|
|
|
|
|
class ImageFormat(Enum):
|
|
RGB565 = 'RGB565'
|
|
RGBA8888 = 'RGBA8888'
|
|
YUYV = 'YUYV'
|
|
CRY1CBY0 = 'CrY1CbY0'
|
|
CBY1CRY0 = 'CbY1CrY0'
|
|
Y1CRY0CB = 'Y1CrY0Cb'
|
|
Y1CBY0CR = 'Y1CbY0Cr'
|
|
CRY0CBY1 = 'CrY0CbY1'
|
|
CBY0CRY1 = 'CbY0CrY1'
|
|
Y0CRY1CB = 'Y0CrY1Cb'
|
|
Y0CBY1CR = 'Y0CbY1Cr'
|
|
RAW8 = 'RAW8'
|
|
YUV420p = 'YUV420p'
|
|
|
|
|
|
class ResizeMode(Enum):
|
|
NONE = 'none'
|
|
ENABLE = 'auto'
|
|
|
|
|
|
class PaddingMode(Enum):
|
|
NONE = 'none'
|
|
PADDING_CORNER = 'corner'
|
|
PADDING_SYMMETRIC = 'symmetric'
|
|
|
|
|
|
class PostprocessMode(Enum):
|
|
NONE = 'none'
|
|
YOLO_V3 = 'yolo_v3'
|
|
YOLO_V5 = 'yolo_v5'
|
|
|
|
|
|
class NormalizeMode(Enum):
|
|
NONE = 'none'
|
|
KNERON = 'kneron'
|
|
TENSORFLOW = 'tensorflow'
|
|
YOLO = 'yolo'
|
|
CUSTOMIZED_DEFAULT = 'customized_default'
|
|
CUSTOMIZED_SUB128 = 'customized_sub128'
|
|
CUSTOMIZED_DIV2 = 'customized_div2'
|
|
CUSTOMIZED_SUB128_DIV2 = 'customized_sub128_div2'
|
|
|
|
|
|
class InferenceRetrieveNodeMode(Enum):
|
|
FIXED = 'fixed'
|
|
FLOAT = 'float'
|