Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
Convolutional neural network model for video classification trained on the Kinetics dataset.
Software that can generate photos from paintings, turn horses into zebras, perform style transfer, and more.
Spatial Temporal Graph Convolutional Networks (ST-GCN) for Skeleton-Based Action Recognition in PyTorch