Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 600 Bytes

ReadMe.md

File metadata and controls

13 lines (9 loc) · 600 Bytes
Raw

This code includes two parts

  1. code including train.py and utils.py
  2. model including d_model.h5 g_model.h5 (generated by model.py)

For Code train.py uses d_model.h5 and g_model.5 for input. Please generate them before training. utils.py includes a module of data loader and implentment of some layers.

Please use python 3.6 and Keras 2.2.

The weights of DTDN can be downloaded at https://drive.google.com/file/d/1s5rrrzdSIOkD017VOcf3Ajq0UpttIzd6/view?usp=sharing

The training data can be downloaded at https://drive.google.com/file/d/1QDNZwLQrxeMFw2qbArNqDwk5bnYFL64G/view?usp=sharing