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