上传者: 42099814
|
上传时间: 2022-06-07 14:57:47
|
文件大小: 15KB
|
文件类型: ZIP
pytorch-polygon-rnn
Pytorch实现。 注意,我使用另一种方法来处理第一个顶点,而不是像本文中那样训练另一个模型。
与原纸的不同
我使用两个虚拟起始顶点来处理第一个顶点,如图像标题所示。
我需要在ConvLSTM层之后添加一个LSTM层,因为我需要输出为D * D + 1维度才能处理结束符号。
如何训练和测试
从下载数据,组织图像文件和注释json文件,如下所示:
img
├── train
│ ├── cityname1
│ │ ├── pic.png
│ │ ├── ...
│ ├── cityname2
│ │ ├── pic.png
│ │ ├── ...
├── val
│ ├── cityname
│ │ ├── pic.png
│ │ ├── ...
├── test
│ ├── ci