rec_train.py训练时准确率一直为0

[wenmuzhou]

rec_train.py训练时准确率一直为0

[terrancraft]

你打印一下网络结构看看对不对

[novioleo]

@hhhhhhogan 抱歉，才看到这个issue。能烦请你上传一下你的对应的图片，以及你的onehot的结果么？

[dengfenglai321]

你好，我这里使用icdar 2017数据集训练，发现准确率一直为0，使用的配置项为rec_train_config.py，没有做什么太大的修改，只是修改了batch_size，alphabet的路径，dataset的路径，训练时log的打印如下：

期间进入到RecMetric.py中查看模型的输出值predictions，发现里面预测的结果都为0,0,0...0的tensor

训练数据的标注如下（图片路径与文字内容中间以\t进行分割）：

E:\DataSets\icdar2017rctw\icdar2017rctw\recognition\train\image_0_0.jpg 金氏眼镜
E:\DataSets\icdar2017rctw\icdar2017rctw\recognition\train\image_0_1.jpg 创于1989
E:\DataSets\icdar2017rctw\icdar2017rctw\recognition\train\image_0_2.jpg 城建店

alphabet中的信息如下，一共5529个字符，加上blank的话那么最后的类别数n_class设置为5530：

另外config中的设置项：
2020-11-05 15:10:33,337 - torchocr - INFO - {'exp_name': 'CRNN', 'train_options': {'resume_from': '', 'third_party_name': '', 'checkpoint_save_dir': './output/CRNN/checkpoint', 'device': 'cuda:0', 'epochs': 200, 'fine_tune_stage': ['backbone', 'neck', 'head'], 'print_interval': 20, 'val_interval': 3000, 'ckpt_save_type': 'HighestAcc', 'ckpt_save_epoch': 4}, 'SEED': 927, 'optimizer': {'type': 'Adam', 'lr': 0.001, 'weight_decay': 0.0001}, 'lr_scheduler': {'type': 'StepLR', 'step_size': 60, 'gamma': 0.1}, 'model': {'type': 'RecModel', 'backbone': {'type': 'ResNet', 'layers': 18}, 'neck': {'type': 'PPaddleRNN'}, 'head': {'type': 'CTC', 'n_class': 5530}, 'in_channels': 3}, 'loss': {'type': 'CTCLoss', 'blank_idx': 0}, 'dataset': {'alphabet': 'E:/pro/ncnn_ocr/models/keys.txt', 'train': {'dataset': {'type': 'RecTextLineDataset', 'file': 'E:/pro/chineseocr-master/train/ocr/txt/icdar2017Backup.txt', 'input_h': 32, 'mean': 0.5, 'std': 0.5, 'augmentation': False}, 'loader': {'type': 'DataLoader', 'batch_size': 4, 'shuffle': True, 'num_workers': 1, 'collate_fn': {'type': 'RecCollateFn', 'img_w': 120}}}, 'eval': {'dataset': {'type': 'RecTextLineDataset', 'file': 'E:/pro/chineseocr-master/train/ocr/txt/2017valBackup.txt', 'input_h': 32, 'mean': 0.5, 'std': 0.5, 'augmentation': False}, 'loader': {'type': 'RecDataLoader', 'batch_size': 4, 'shuffle': False, 'num_workers': 1, 'collate_fn': {'type': 'RecCollateFn', 'img_w': 120}}}}}

希望得到回复，谢谢！

你好，，请问你解决这个问题了吗？我也遇到了这个问题

[luotianhang]

您好，我也遇到了这个问题，请问您是怎么解决的

[wzl639]

我也是这个问题

[novioleo]

重新拉取代码，并升级下pytorch的版本，到1.7试试。
@wzl639

[wzl639]

重新拉取代码，并升级下pytorch的版本，到1.7试试。
@wzl639
感谢解答 但是我的torch是1.7

[wzl639]

重新拉取代码，并升级下pytorch的版本，到1.7试试。
@wzl639
感谢解答 但是我的torch是1.7

博主我还有一个问题 单独发出来 能帮我看看吗

[luotianhang]

我做过实验 你要训练到很后面才会开始有效果

[Tomhardy13]

我做过实验 你要训练到很后面才会开始有效果

请问一般要多久才开始有效果？@luotianhang