Closed Chmoqq closed 4 years ago
Traceback (most recent call last): File "/Users/ivandaniyelyan/nomeroff-net/train/py/options_train.py", line 41, in npClassificator.prepare(PATH_TO_DATASET, verbose=1) File "/Users/ivandaniyelyan/nomeroff-net/NomeroffNet/OptionsDetector.py", line 150, in prepare self.train_generator = self.compile_train_generator(train_dir, (self.HEIGHT, self.WEIGHT), self.BATCH_SIZE) File "/Users/ivandaniyelyan/nomeroff-net/NomeroffNet/OptionsDetector.py", line 345, in compile_train_generator imageGenerator = ImgGenerator(train_dir, self.WEIGHT, self.HEIGHT, self.BATCH_SIZE, [len(self.CLASS_STATE), len(self.CLASS_REGION), len(self.CLASS_COUNT_LINE)]) File "/Users/ivandaniyelyan/nomeroff-net/NomeroffNet/Base/ImgGenerator.py", line 33, in init self.samples.append([img_filepath, [description["state_id"], description["region_id"], description["count_lines"]]]) KeyError: 'count_lines'
Загрузил датасет с сайта для тренировки модели, в json файлах отсутствует параметр count_lines Может что-то не так делаю Буду благодарен за ответ
Traceback (most recent call last): File "/Users/ivandaniyelyan/nomeroff-net/train/py/options_train.py", line 41, in
npClassificator.prepare(PATH_TO_DATASET, verbose=1)
File "/Users/ivandaniyelyan/nomeroff-net/NomeroffNet/OptionsDetector.py", line 150, in prepare
self.train_generator = self.compile_train_generator(train_dir, (self.HEIGHT, self.WEIGHT), self.BATCH_SIZE)
File "/Users/ivandaniyelyan/nomeroff-net/NomeroffNet/OptionsDetector.py", line 345, in compile_train_generator
imageGenerator = ImgGenerator(train_dir, self.WEIGHT, self.HEIGHT, self.BATCH_SIZE, [len(self.CLASS_STATE), len(self.CLASS_REGION), len(self.CLASS_COUNT_LINE)])
File "/Users/ivandaniyelyan/nomeroff-net/NomeroffNet/Base/ImgGenerator.py", line 33, in init
self.samples.append([img_filepath, [description["state_id"], description["region_id"], description["count_lines"]]])
KeyError: 'count_lines'
Загрузил датасет с сайта для тренировки модели, в json файлах отсутствует параметр count_lines Может что-то не так делаю Буду благодарен за ответ