FelixOpolka/STGCN-PyTorchPublic Notifications Fork67 Star315 Code Issues8 Pull requests Actions Projects Security Insights Additional navigation options Files master data .gitignore LICENSE README.md main.py stgcn.py utils.py
OracleRay/STGCN_pytorchmain 1 BranchTags Code This branch is 2 commits ahead of hazdzz/STGCN:main.Folders and filesLatest commit OracleRay 注释调没用的代码 51c7b5e· Oct 15, 2024 History229 Commits data simplify the code Feb 8, 2022...
train_model.pyandtrain_main.py: codes for training models ./test/ testing_model.py: codes for evaluating models ./utils/: containing useful files that are required in the project (e.g. data loader, metrics calculation, loss function, configurations) ...
本项目推理模型权重采用Github仓库中Pytorch框架的STGCN模型训练SZ-Taxi数据集得到的权重转化得到。经过以下两步模型转化: 1、pth转化为onnx 可以根据实际的路径和输入大小修改export_onnx.py(该文件需要依赖于项目结构目录,请放到训练代码项目主目录下再运行)运行指令如下: python export_onnx.py 转换好...
The PyTorch version of STGCN. Contribute to lensfa-lzd/STGCN development by creating an account on GitHub.
227 Commits data figure model script .gitignore LICENSE README.md main.py requirements.txt README LGPL-2.1 license Spatio-Temporal Graph Convolutional Networks About The PyTorch implementation of STGCN from the paperSpatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Fore...
Official PyTorch Implementation of the paper:Context-based Interpretable Spatio-Temporal Graph Convolutional Network for Human Motion Forecastingaccepted inWACV2024 [Proceedings] [Arxiv] Installation To setup the environment: cd cistgcn conda create -n cistgcn python=3.10 ...
The implementation of Spatio-Temporal Graph Convolutional Layer with PyTorch is available inPyG Temporal. You might refer toSTConvthat supports ChebConv Graph Convolutions. Dataset PeMSD7was collected from Caltrans Performance Measurement System (PeMS) in real-time by over 39, 000 sensor stations, dep...
This is a pytorch implementation of our paper "Unified Spatio-Temporal Modeling for Traffic Forecasting using Graph Neural Network" which has been accepted by IJCNN 2021. Check the video presentation of our paper here. Abstract Research in deep learning models to forecast traffic intensities has gain...
main.py 更新 Mar 31, 2023 nni_experiment.py 调整实验细节 Nov 1, 2022 Repository files navigation README A PyTorch Version of STGCN Base on hazdzz/STGCN with nni for Hyperparameter Optimization Paper Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting. ...