更新文档

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()