pretrained

[06Liz]

const.py文件中，import os
from pathlib import Path

Use environment variables to auto-detect whether we are running an a Compute Canada cluster:

Thanks to https://github.com/DM-Berger/unet-learn/blob/master/src/train/load.py for this trick.

COMPUTECANADA = False
TMP = os.environ.get("SLURM_TMPDIR")

if TMP:
COMPUTECANADA = True

if COMPUTECANADA:
INPUT_FOLDER = Path(str(TMP)).resolve() / "work" / "inputs"
MASK_FOLDER = Path(str(TMP)).resolve() / "work" / "inputs" / "masks"
PRETRAINED_MODEL_FOLDER = Path(str(TMP)).resolve() / "work" / "trained_models"
PRETRAINED_MODEL_DDPM_PATH = (
Path(str(TMP)).resolve() / "work" / "trained_models" / "ddpm"
)
PRETRAINED_MODEL_VAE_PATH = (
Path(str(TMP)).resolve() / "work" / "trained_models" / "vae"
)
PRETRAINED_MODEL_DECODER_PATH = (
Path(str(TMP)).resolve() / "work" / "trained_models" / "decoder"
)
PRETRAINED_MODEL_VGG_PATH = (
Path(str(TMP)).resolve() / "work" / "trained_models" / "vgg16.pt"
)
OUTPUT_FOLDER = Path(str(TMP)).resolve() / "work" / "outputs"
else:
INPUT_FOLDER = Path(file).resolve().parent.parent.parent / "data" / "IXI"
MASK_FOLDER = Path(file).resolve().parent.parent / "masks"
OASIS_FOLDER = Path(file).resolve().parent.parent.parent / "data" / "OASIS"
PRETRAINED_MODEL_FOLDER = (
Path(file).resolve().parent.parent.parent / "data" / "trained_models"
)这些预训练的模型都在哪里啊，数据在哪啊？

[06Liz]

而且那个链接也打不开

[jueqiw]

Thanks for your interest in our work! The weight for the pre-trained model could be found in https://drive.google.com/drive/folders/110l68um6gUJzECIv0AyF-4Fcw0rrQgA9?usp=drive_link

Please don't hesitate to let me know if you have future questions!

[06Liz]

Thanks for your answer and your contribution.

[06Liz]

with open(
    "/scratch/j/jlevman/jueqi/thesis_experiments/decoder/result_decoder_downsample_2.csv",
    "a",
) as file:
    writer = csv.writer(file)
    writer.writerow(row)

return latent_vector_out

@wendy-xiaozong
Where is the /scratch/j/jlevman/jueqi/thesis_experiments/decoder/result_decoder_downsample_2.csv?

[jueqiw]

It's an output file that I used to save the result