コード例 #1
0
    def __init__(self):
        OptionsDetector.__init__(self)
        # outputs 1
        self.CLASS_STATE = ["BACKGROUND", "FILLED", "NOT_FILLED"]

        # outputs 2
        self.CLASS_REGION = [
            "xx-unknown", "eu-ua-2015", "eu-ua-2004", "eu-ua-1995", "eu",
            "xx-transit"
        ]

        self.EPOCHS = 2
        self.BATCH_SIZE = 10

        self.HEIGHT = 64
        self.WEIGHT = 295
コード例 #2
0
ファイル: options_train.py プロジェクト: syn4ps/nomeroff-net
    def __init__(self):
        OptionsDetector.__init__(self)
        # outputs 1
        self.CLASS_STATE = ["garbage", "filled", "not filled", "empty"]

        # outputs 2
        self.CLASS_REGION = [
            "xx-unknown", "eu-ua-2015", "eu-ua-2004", "eu-ua-1995", "eu",
            "xx-transit", "ru", "kz", "eu-ua-ordlo-dnr", "eu-ua-ordlo-lnr",
            "ge"
        ]

        # output 3
        self.CLASS_COUNT_LINE = ["0", "1", "2"]

        self.EPOCHS = 100
        self.BATCH_SIZE = 1  # 64

        self.HEIGHT = 64
        self.WEIGHT = 295