Skip to content

Instantly share code, notes, and snippets.

import torch
import torch.nn as nn
import torch.nn.functional as F
import copy
from tqdm import trange
passed = torch.zeros(50)
for i in trange(50):
pad = tuple(list(torch.randint(3,(3,))))
dilation = tuple(list(torch.randint(2,(3,))+1))
import torch
import torch.nn as nn
import torch.nn.functional as F
import nibabel as nib
import time
import numpy as np
import scipy.ndimage
from scipy.ndimage.interpolation import zoom as zoom
from scipy.ndimage.interpolation import map_coordinates
import torch
import torch.nn as nn
import torch.nn.functional as F
import nibabel as nib
import time
import numpy as np
import scipy.ndimage
from scipy.ndimage.interpolation import zoom as zoom
from scipy.ndimage.interpolation import map_coordinates
@mattiaspaul
mattiaspaul / adamreg_mind.py
Last active July 29, 2023 20:59
AdamReg MIND
#note: an advanced baseline with additional infos for registration settings
#for Learn2Reg 2021 and recommended pre-processing of lung scans can be found
#here https://github.com/multimodallearning/convexAdam
import torch
import torch.nn as nn
import torch.nn.functional as F
import nibabel as nib
import time
import numpy as np
import scipy.ndimage
@mattiaspaul
mattiaspaul / landmarks_split_txt.py
Created July 10, 2018 17:54
convert MINC landmark tag file into two txt-files readable for c3d landmark-to-spheres
import numpy as np
import argparse
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--inputtag", dest="inputtag", help="input tag file from (MINC)", default=None, required=True)
parser.add_argument("--savetxt", dest="savetxt", help="output landmark file to (txt)", default=None, required=True)
@mattiaspaul
mattiaspaul / landmarks_centre_mass.py
Created July 10, 2018 17:51
calculate centre of mass for 3D nifti segmentation mask and TRE
import numpy as np
import nibabel as nib
import argparse
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--inputnii", dest="inputnii", help="input segmentation from (nii.gz)", default=None, required=True)
parser.add_argument("--movingnii", dest="movingnii", help="second segmentation from (nii.gz)", default=None, required=False)
parser.add_argument("--savetxt", dest="savetxt", help="output landmark file to (txt)", default=None, required=False)