Tsingzao's repositories
Graphormer
Graphormer is a deep learning package that allows researchers and developers to train custom models for molecule modeling tasks. It aims to accelerate the research and application in AI for molecule science, such as material discovery, drug discovery, etc.
Prediction-of-radar-precipitation-and-wind-speed
Prediction of radar, precipitation and wind speed
predrnn-pytorch
Official implementation for NIPS'17 paper: PredRNN: Recurrent Neural Networks for Predictive Learning Using Spatiotemporal LSTMs.
PyTorch-GAN
PyTorch implementations of Generative Adversarial Networks.
pytorch-generative
Easy generative modeling in PyTorch.
climate-learn
Source code for ClimateLearn
darts
A python library for easy manipulation and forecasting of time series.
datasets
Datasets for deep learning with satellite & aerial imagery
deep-time-series
Deep learning PyTorch library for time series forecasting
detectron2
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
earth-forecasting-transformer
Official implementation of Earthformer
GNN4Traffic
This is the repository for the collection of Graph Neural Network for Traffic Forecasting.
mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
ncl
The NCAR Command Language (NCL) is a scripting language for the analysis and visualization of climate and weather data.
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
radar_nowcasting
A DEMO of weather radar nowcasting using deep learning model (SmaAt-Unet)
segmentation_models.pytorch
Segmentation models with pretrained backbones. PyTorch.
SmaAt-UNet
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
STG-NCDE
"Graph Neural Controlled Differential Equations for Traffic Forecasting", AAAI 2022
vp-suite
A general framework for video prediction in PyTorch.
wrf-python
A collection of diagnostic and interpolation routines for use with output from the Weather Research and Forecasting (WRF-ARW) Model.