From 9dedc81912710f9ce6efe2e04711fad1b69e6812 Mon Sep 17 00:00:00 2001 From: sml2h3 Date: Sat, 26 Feb 2022 15:15:47 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=96=87=E6=A1=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- utils/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/utils/train.py b/utils/train.py index 752b4e6..04fa0aa 100644 --- a/utils/train.py +++ b/utils/train.py @@ -33,6 +33,7 @@ class Train: self.resize = [int(self.conf['Model']['ImageWidth']), int(self.conf['Model']['ImageHeight'])] self.word = self.conf['Model']['Word'] + self.ImageChannel = self.conf['Model']['ImageChannel'] logger.info("\nTaget:\nmin_Accuracy: {}\nmin_Epoch: {}\nmax_Loss: {}".format(self.target_acc, self.min_epoch, self.max_loss)) @@ -147,7 +148,7 @@ class Train: time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime(self.now_time)))) , input_names, output_names, dynamic_ax) with open(os.path.join(self.models_path, "charsets.json"), 'w', encoding="utf-8") as f: - f.write(json.dumps({"charset": self.net.charset, "image": self.resize, "word": self.word}, ensure_ascii=False)) + f.write(json.dumps({"charset": self.net.charset, "image": self.resize, "word": self.word, 'channel': self.ImageChannel}, ensure_ascii=False)) logger.info("\nExport Finished!Using Time: {}min".format( str(int(int(self.now_time) - int(self.start_time)) / 60))) exit()