S³NET

Pytorch implementation of "S³NET: GRAPH REPRESENTATIONAL NETWORK FOR SKETCH RECOGNITION"
It has been accepted by ICME2020.

Recognition Result

Prerequisites

• Linux (tested on Ubuntu 16.04)
  
• Pytorch >= 1.2
  
• NVIDIA GPU + CUDA CuDNN
  
• torch_geometric PyG
  

Dataset

• Sketch-RNN QuickDraw Dataset Download

Conclusion

Thank you and sorry for the bugs! If you would have further discussion on this code repository, please feel free to send email to LAN YANG. Email: [email protected]