diff --git a/ppocr/modeling/transforms/tps.py b/ppocr/modeling/transforms/tps.py index de4bb7a686b0c8ff838338ffe51bb0ea1952057c..81221b0351944ef8a054d44d3857a8c0ad61e052 100644 --- a/ppocr/modeling/transforms/tps.py +++ b/ppocr/modeling/transforms/tps.py @@ -326,6 +326,4 @@ class STN_ON(nn.Layer): image, self.tps_inputsize, mode="bilinear", align_corners=True) stn_img_feat, ctrl_points = self.stn_head(stn_input) x, _ = self.tps(image, ctrl_points) - #print("x:", np.sum(x.numpy())) - # print(x.shape) return x diff --git a/tools/program.py b/tools/program.py index 3479ff26fc3cfab39e658b03f626333100f8ed74..f77c69f88f50a4b4762b8003dc3efbd8580c3faa 100755 --- a/tools/program.py +++ b/tools/program.py @@ -215,9 +215,6 @@ def train(config, preds = model(images, data=batch[1:]) else: preds = model(images) - state_dict = model.state_dict() - # for key in state_dict: - # print(key) loss = loss_class(preds, batch) avg_loss = loss['loss'] avg_loss.backward() @@ -414,7 +411,6 @@ def preprocess(is_train=False): yaml.dump( dict(config), f, default_flow_style=False, sort_keys=False) log_file = '{}/train.log'.format(save_model_dir) - print("log has save in {}/train.log".format(save_model_dir)) else: log_file = None logger = get_logger(name='root', log_file=log_file) diff --git a/tools/train.py b/tools/train.py index e1515f57c9402adf1f7d00c03aee78ea20c2a371..20f5a670d5c8e666678259e0042b3b790e528590 100755 --- a/tools/train.py +++ b/tools/train.py @@ -72,8 +72,6 @@ def main(config, device, logger, vdl_writer): # for rec algorithm if hasattr(post_process_class, 'character'): char_num = len(getattr(post_process_class, 'character')) - character = getattr(post_process_class, 'character') - print("getattr character:", character) if config['Architecture']["algorithm"] in ["Distillation", ]: # distillation model for key in config['Architecture']["Models"]: