├── README.md ├── checkpoints └── model_x8.pth ├── data └── Set5 │ ├── img_001.bmp │ ├── img_002.bmp │ ├── img_003.bmp │ ├── img_004.bmp │ └── img_005.bmp ├── dataset.py ├── main.py ├── model6b.py ├── model8b8x.py ├── model8b8xt.py ├── pyssim.py ├── test.sh ├── test1se.py └── train.sh /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/model_x8.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/checkpoints/model_x8.pth -------------------------------------------------------------------------------- /data/Set5/img_001.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/data/Set5/img_001.bmp -------------------------------------------------------------------------------- /data/Set5/img_002.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/data/Set5/img_002.bmp -------------------------------------------------------------------------------- /data/Set5/img_003.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/data/Set5/img_003.bmp -------------------------------------------------------------------------------- /data/Set5/img_004.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/data/Set5/img_004.bmp -------------------------------------------------------------------------------- /data/Set5/img_005.bmp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/data/Set5/img_005.bmp -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/dataset.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/main.py -------------------------------------------------------------------------------- /model6b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/model6b.py -------------------------------------------------------------------------------- /model8b8x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/model8b8x.py -------------------------------------------------------------------------------- /model8b8xt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/model8b8xt.py -------------------------------------------------------------------------------- /pyssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/pyssim.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/test.sh -------------------------------------------------------------------------------- /test1se.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/opteroncx/TAN/HEAD/test1se.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0,1,2 python3 main.py --cuda --nEpochs 100 --------------------------------------------------------------------------------