From 44ac3db79fd783d80915a3773ab82d6788174ed6 Mon Sep 17 00:00:00 2001 From: sml2h3 Date: Sat, 26 Feb 2022 15:09:54 +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 | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/utils/train.py b/utils/train.py index 795bbbd..9905dba 100644 --- a/utils/train.py +++ b/utils/train.py @@ -30,6 +30,9 @@ class Train: self.target_acc = self.target['Accuracy'] self.min_epoch = self.target['Epoch'] self.max_loss = self.target['Cost'] + + self.resize = [int(self.conf['Model']['ImageWidth']), int(self.conf['Model']['ImageHeight'])] + logger.info("\nTaget:\nmin_Accuracy: {}\nmin_Epoch: {}\nmax_Loss: {}".format(self.target_acc, self.min_epoch, self.max_loss)) @@ -143,8 +146,8 @@ class Train: self.project_name, str(accuracy), self.epoch, self.step, 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, "charset.json"), 'w', encoding="utf-8") as f: - f.write(json.dumps(self.net.charset, ensure_ascii=False)) + 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]}, ensure_ascii=False)) logger.info("\nExport Finished!Using Time: {}min".format( str(int(int(self.now_time) - int(self.start_time)) / 60))) exit()