MinistTHZ

MinistTHZ Dataset will public soon!

packet requrement python==3.6.7 pytorch==1.2.0

train: python PM_train.py

test: python PM_test.py

we also relese the code of the Unet/esrgan/transformer for the abution study.

Please contact us in the future regarding the use of private datasets.

In the future we will further develop our experimental section and the datasets in this repositorie.

Name		Name	Last commit message	Last commit date
Latest commit History 21 Commits
cupy_layers		cupy_layers
data/train		data/train
layers		layers
logdir/train.events		logdir/train.events
models/COTNet+PMANN34.789505_0.996199_150000cos		models/COTNet+PMANN34.789505_0.996199_150000cos
pytorch-gradual-warmup-lr		pytorch-gradual-warmup-lr
swin_transformer_pytorch		swin_transformer_pytorch
test		test
DualGCN.py		DualGCN.py
ESRGAN.py		ESRGAN.py
ExternalAttention.py		ExternalAttention.py
GCN_predict.py		GCN_predict.py
GCN_train.py		GCN_train.py
MPRNet.py		MPRNet.py
PM_predict.py		PM_predict.py
PM_train.py		PM_train.py
README.md		README.md
backbone.py		backbone.py
blocks.py		blocks.py
cal_ssim.py		cal_ssim.py
calculate_PSNR_SSIM.py		calculate_PSNR_SSIM.py
dataset.py		dataset.py
edcnn_model.py		edcnn_model.py
esrgan_predict.py		esrgan_predict.py
esrgan_train.py		esrgan_train.py
involution.py		involution.py
losses.py		losses.py
mprnet_predict.py		mprnet_predict.py
multi_head_attention_torch.py		multi_head_attention_torch.py
plotLoss.py		plotLoss.py
pmann.py		pmann.py
run.sh		run.sh
settings.py		settings.py
swintransformer_predict.py		swintransformer_predict.py
unet_model.py		unet_model.py
unet_predict.py		unet_predict.py
unet_train.py		unet_train.py

Provide feedback