/skylark

Astro project

Primary LanguagePythonMIT LicenseMIT

skylark

Astro project, interframe prediction of protoplanetary disk using resnet.

Protoplanetary disk

Requirements

cuda9.0.176 cudnn7.1/7.5

Python requirements:

  • pytorch
  • numpy
  • matplotlib
  • h5py
  • tensorboardX
  • Future
  • Pillow
  • opencv-python
  • scipy

Usage

  • Save raw data to h5py files

python main.py -m save

  • Train model

python main.py

  • Visualize

python main.py -m v

  • Clean up file/directory from last run

python main.py -m c