运行vis_pred.py 时报错 about hdmapnet

[Yc174]

请问能公布一下训练好的model吗？ @2443138933 @kd12138888

[kd12138888]

coords, _, _ = vectorize(segmentation[si], embedding[si], direction[si], angle_class)
TypeError: 'NoneType' object is not subscriptable
请问如何解决？

[kd12138888]

segmentation, embedding, direction = model(imgs.cuda(), trans.cuda(), rots.cuda(), intrins.cuda(),
post_trans.cuda(), post_rots.cuda(), lidar_data.cuda(),
lidar_mask.cuda(), car_trans.cuda(), yaw_pitch_roll.cuda())
其中embedding参数为空

[2443138933]

--instance_seg和 --direction_pred都改成True

[kd12138888]

parser.add_argument('--instance_seg', action='store_true')
parser.add_argument('--direction_pred', action='store_true')
是这两个吗？ 一开始就是ture

[2443138933]

parser.add_argument('--instance_seg', action='store_true') parser.add_argument('--direction_pred', action='store_true') 是这两个吗？ 一开始就是ture

action='store_true'改成default=True

[kd12138888]

非常感谢