├── README.md ├── code ├── dataset.py ├── model.py ├── torch_utils.py ├── train.py └── utils.py ├── data └── note.txt └── fig ├── architecture.png └── performance.png /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/README.md -------------------------------------------------------------------------------- /code/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/code/dataset.py -------------------------------------------------------------------------------- /code/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/code/model.py -------------------------------------------------------------------------------- /code/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/code/torch_utils.py -------------------------------------------------------------------------------- /code/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/code/train.py -------------------------------------------------------------------------------- /code/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/code/utils.py -------------------------------------------------------------------------------- /data/note.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /fig/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/fig/architecture.png -------------------------------------------------------------------------------- /fig/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenguonie/2020_StressGAN/HEAD/fig/performance.png --------------------------------------------------------------------------------