Open wangyan-hlab opened 1 week ago
你好,
要应用于三维 TSP,需要修改的主要是两个部分:
State 和 Environment 相关的修改
在 state
和 environment
中,有许多与距离计算和 action 执行相关的代码,这些都与 TSP 问题的维度相关,因此需要做出相应的修改。
模型的 Encoder 修改
当前使用的 encoder 参考了 3D 点云的处理方式来对大规模 TSP 问题进行编码。对于三维 TSP 问题,可以考虑直接使用针对 3D 点云设计的 encoder 结构。例如,可以参考以下文献中的 PointPillars 结构来进行修改:
文献:Lang, A. H.; Vora, S.; Caesar, H.; Zhou, L.; Yang, J.; and Beijbom, O. 2019. PointPillars: Fast Encoders for Object Detection From Point Clouds. In 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 12689–12697.
相关代码段:代码链接
@neo-pan 你好,
如果我想将这个算法应用于三维TSP问题(顶点位于三维空间中),需要进行哪些修改呢?
谢谢!