mrswin / graph-unet-traffic-prediction Goto Github PK
View Code? Open in Web Editor NEWThis project forked from lucahermes/graph-unet-traffic-prediction
Hybrid UNet model for traffic prediction from traffic movies. The hybrid graph operation is a mixture of CNN and GNN operations to capture pixel topology and improve spatial generalization.