GithubHelp home page GithubHelp logo

chao-tan / forecast-clstm Goto Github PK

View Code? Open in Web Editor NEW
10.0 1.0 0.0 676 KB

Pytorch implementation of paper: FORECAST-CLSTM: A New Convolutional LSTM Network for Cloudage Nowcasting (VCIP2018).

License: MIT License

Python 100.00%
convolutional-lstm-network deep-learning cloudage-nowcasting lstm new-dataset forecast-clstm scmd-dataset

forecast-clstm's Introduction

FORECAST-CLSTM: A New Convolutional LSTM Network for Cloudage Nowcasting

by Chao Tan, Xin Feng, Jianwu Long, and Li Geng.

This repository contains the source code and dataset for FORECAST-CLSTM, provided by Chao Tan.

The paper is avaliable for download here. Click here for more details.


Dataset

Our SCMD2016 dataset is available for download at TianYiCloud(2.5GB) or BaiduCloud(2.5GB) (extraction code: ssby).
SCMD dataset is a brand new cloudage nowcasting dataset for deep learning research. It contains 20000 grayscale image sequences for training and another 3500 image sequences for testing. You can get the SCMD2016 dataset at any time but only for scientific research. At the same time, please cite our work when you use the SCMD dataset.

The mnist dataset in npz format can be download here.

Prerequisites

  • Python 3.5
  • PyTorch >= 0.4.0
  • opencv 0.4
  • PyQt 4
  • numpy

Train

  1. For moving-mnist training, please download mnist.npz dataset and place it in ./data folder, for cloudage nowcasting training, please download and unzip SCMD2016 dataset and place it in ./data folder.
  2. For moving-mnist dataset, run python trainer_mnist.py --model "FORECAST_CLSTM_M" --epochs 100 --train-batch 16 --gpu-ids "0" --checkpoint "checkpoint/forecast_clstm_m" to start training.
  3. For scmd2016 dataset, run python trainer_scmd.py --model "FORECAST_CLSTM_S" --epochs 100 --train-batch 16 --gpu-ids "0" --checkpoint "checkpoint/forecast_clstm_s"" to start training.

Citation

@inproceedings{
     title={{FORECAST-CLSTM}: A New Convolutional LSTM network for Cloudage Nowcasting},
     author={Tan, Chao and Feng, Xin and Long, Jianwu and Geng, Li},
     booktitle={VCIP},
     year={2018},
     note={to appear},
}

forecast-clstm's People

Contributors

chao-tan avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar

forecast-clstm's Issues

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.