环境python3.6 pytorch1.0.0 cuda9.0
数据集说明:https://tianchi.aliyun.com/competition/entrance/231648/information
数据集下载链接:https://pan.baidu.com/s/1ZXzHCdbfTuegjSv8IFewOQ 密码:lhx5
只需下载 train1.rar和tarain2.rar
下载后解压 把两个文件夹里融合,命名为Image
如图
下载标注信息和源码https://pan.baidu.com/s/179I2QaVIqYw0BqQHn2GU7w r6nk
下载完成后
在train/train_lm.py内修改标注和图片的路径地址。
运行train/test_lm.py可视化结果
相关论文https://arxiv.org/abs/1711.07319