quancore / social-lstm

Social LSTM implementation in PyTorch
410 stars 248 forks source link

Question about grid #34

Open benemana opened 4 years ago

benemana commented 4 years ago

Please, could you explain what the parameter "is_occupancy" stands for in grid.py? With respect to the original implementation, I saw you implemented two different kinds of mask, according to the "is_occupancy" value. What does it represent? Why the two masks have different shape?

Thank you a lot.