Skip to content

ziqicui001/GNN_based-Perception-Prediction-Model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Perception Prediction Model

License Build Status Alt text

Contents

Introduction

This study explores whether incorporating the spatial relationships and geometric features of individual street elements can enhance the accuracy and interpretability of perception prediction models. We used quantified spatial and geometric features of urban street elements as input for a graph neural network (GNN) model, a deep learning approach designed to process unstructured data. Compared to traditional methods, the GNN model achieved approximately a 6.3% on average improvement in accuracy across six perception dimensions, highlighting the importance of integrating spatial and geometric features into predictive modeling. Additionally, the study trained an explainer to analyze prediction results, enabling precise identification of critical subgraphs within images, which can determine the key edges, node combinations, and node features that significantly influence perception scores. Compared to previous approaches, this method provides more granular explanation results, offering valuable insights for urban planners to design environments that better align with public perceptions.

Overview

model_result

Instructions

This project has three main step: data processing, model training and explainer training.

Environment Configuration:

CUDA Version: 12.5
Python 3.10
opencv-python 4.10.0.84
openpyxl 3.1.2
openvino 2024.4.0
tensorflow 2.6.0
tensorflow-estimator 2.6.0
torch 2.4.0
torch_geometric 2.5.3

Acknowledgements

Thanks to my collaborators Shangyu Lou. She has made great contributions to methodology, data processing and model training.

Contact

If you have any question, be free to contact me: ziqi.cui@polimi.it

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors