BiTraP:Bi-directional Pedestrian Trajectory Prediction with Multi-modal Goal Estimation
importtorchimporttorch.nnasnnimporttorch.nn.functionalasF#先验网络classPrior(nn.Module):def__init__(self,input_size=256,output_size=64):super(Prior,self).__init__()self.input_size=input_size#输入大小self.outp