Skip to content

CorPiR v1.0.0

Latest

Choose a tag to compare

@ZHT150798 ZHT150798 released this 08 Aug 16:23

Pretrained weights and the test set for CorPiR.

File Size Purpose
corpir_testset.zip 3.9 MB the 122 high-resolution test patches used in the paper (hr_128/); serves all three up-sampling factors
liif_edsr_pathology.pth.tar 25 MB first-stage LIIF encoder, required for every scale
corpir_x2_gen.pth 423 MB diffusion weights, x2
corpir_x4_gen.pth 444 MB diffusion weights, x4
corpir_x8_gen.pth 527 MB diffusion weights, x8
corpir_x2_opt.pth 721 MB optimizer state, x2 - only needed to resume or fine-tune training
corpir_x4_opt.pth 763 MB optimizer state, x4
corpir_x8_opt.pth 930 MB optimizer state, x8

For inference you only need liif_edsr_pathology.pth.tar and the _gen.pth file for the
scale you want. Put them in pretrained/ and unpack corpir_testset.zip into your dataset
folder. See the README for details.