D641593 / MixNet

MIT License
60 stars 9 forks source link

训练和评估total_text遇到的问题 #20

Open CookieKIKI opened 6 months ago

CookieKIKI commented 6 months ago

首先,感谢您超棒的作品!

其次,我是一个新手,可能问题有些多!

我在使用您的网络训练并评估total-text遇到了一些问题: 1、训练损失到0.38左右就不收敛了; 2、不知道如何获得P、R的值; 以下是我的操作流程: 训练:CUDA_VISIBLE_DEVICES=0 python3 train_mixNet.py --exp_name Totaltext --net FSNet_S --batch_size 4--input_size 640 --max_epoch 600 --optim Adam --lr 0.0005 --num_workers 4 评估:python3 eval_mixNet.py --net FSNet_S --scale 1 --exp_name Totaltext --checkepoch 600 --test_size 640 1024 --dis_threshold 0.3 --cls_threshold 0.8(这一步输出为大量txt文件,作为下一步输入); 测试:采用readme中TIoU-metric-python3获取p、r值,输入代码:python curved-tiou/script.py -g=total-text-gt.zip -s=totaltext_eval.zip 结果出现错误: Error ! (AssertionError('polygon has intersection sides', [(103.0,63.9),(987.8,70.0), (986.8,86.8), (987.0, 10.), (993.8, 108.0), (105.8, 116.8), (1012.8, 12 .B), (010.8; 188), (1016.0, 60.8), (102.8,2.8), (186.0, 8.0), (108.8, 77.8), (1028.8, 95.9),(1041.8, 93.0), (1047.8,78.0), (1043., 76.), (1037.8, 7 ), (1032.8, 72.8), (1028.8, 71.0), (1018.0, 61.0)1)), b"Line in sample not valid. sample: 1 Line: 103,63,987,78 986,86,987,100,93,188, 1085 116,1B12,121 10 8.18, 1016,68,121,2,1096,8,198,77,1028,95,1041,93,1847.78,1043,76,1837,73,103272,1028,71,1818,61 Epror: ('polvgon has intersection sides'. [(1883.8. 63.0). (987.0, 70.0), (986.8, 86.8), (987.0, 108.8),(99.0, 108.8), (1005.0, 16.8), (10128,121.0), (1010.8, 118.0), (116.0, 60.0), (1021.0, 2.0), (1B6.0, B.0) (1088.0, 77.8); (1028.0, 95.0); (1041.0, 93.8); (1047.0, 78.0); (1043.0, 76.8); (137.8,73.0); (1032.8, 72.8); (1028.8, 71.8); (1018.8, 61.)])")

是不是我哪里没有理解到位,导致结果出现问题?麻烦有人可以给予解答!非常感谢!!!

CookieKIKI commented 5 months ago

抱歉,是我的失误,已经解决p、r值的获取问题,但是可能由于训练时不收敛的问题,精度不够理想,请问如何调整?