/Fault-Diagnosis-of-Wheeled-Robot

Fault Diagnosis of Wheeled Robot Based on Prior Knowledge and Spatial-temporal difference graph convolutional network

Primary LanguagePython

Spatial-temporal Difference Graph Convolutional Network (STDGCN)

This code is about the implementation of Fault Diagnosis of Wheeled Robot Based on Prior Knowledge and Spatial-temporal Difference Graph Convolutional Network.

Note

The framework of this code is based on STGCN-PyTorch.

Requirements

  • Torch == 1.9.0
  • Torchvision == 0.10.0
  • Numpy