Jialiang123
Jialiang123
继续训练
恢复训练时,除了在config.py中更改三行 pretrained = True restart_training = False checkpoint = '/home/kpl/PSENet/outputs/13.pth' 如果有如下错误 RuntimeError: Error(s) in loading state_dict for PSENet:........................................ 可尝试在dataset/data_utils.py中  将load_state_dict(state_dict) 改成 model.load_state_dict(state_dict, False)
批量测试图片
各位老师,想请问一下怎么样可以快速批量测试图片呀,我用自己的数据集训练好了模型,这会想用图片批量测试一下效果,现在的demo.py只可以测试单张,写for循环测试速度太慢了,只有图片数据,没有标注文件,恳请给点建议,谢谢
Thanks, this is a great work. When can I see the Evaluation Code