def __init__()

in data/dryice1.py [0:0]

• 38 lines of code
• 8 McCabe index (conditional complexity)

    def __init__(self, camerafilter, framelist, keyfilter,
            fixedcameras=[], fixedcammean=0., fixedcamstd=1.,
            imagemean=0., imagestd=1.,
            worldscale=1., subsampletype=None, subsamplesize=0):
        krtpath = "experiments/dryice1/data/KRT"
        krt = load_krt(krtpath)

        # get options
        self.allcameras = sorted(list(krt.keys()))
        self.cameras = list(filter(camerafilter, self.allcameras))
        self.framelist = framelist
        self.framecamlist = [(x, cam)
                for x in self.framelist
                for cam in (self.cameras if len(self.cameras) > 0 else [None])]

        self.keyfilter = keyfilter
        self.fixedcameras = fixedcameras
        self.fixedcammean = fixedcammean
        self.fixedcamstd = fixedcamstd
        self.imagemean = imagemean
        self.imagestd = imagestd
        self.subsampletype = subsampletype
        self.subsamplesize = subsamplesize

        # compute camera positions
        self.campos, self.camrot, self.focal, self.princpt = {}, {}, {}, {}
        for cam in self.cameras:
            self.campos[cam] = (-np.dot(krt[cam]['extrin'][:3, :3].T, krt[cam]['extrin'][:3, 3])).astype(np.float32)
            self.camrot[cam] = (krt[cam]['extrin'][:3, :3]).astype(np.float32)
            self.focal[cam] = (np.diag(krt[cam]['intrin'][:2, :2]) / 4.).astype(np.float32)
            self.princpt[cam] = (krt[cam]['intrin'][:2, 2] / 4.).astype(np.float32)

        # transformation that places the center of the object at the origin
        transfpath = "experiments/dryice1/data/pose.txt"
        self.transf = np.genfromtxt(transfpath, dtype=np.float32, skip_footer=2)
        self.transf[:3, :3] *= worldscale

        # load background images for each camera
        if "bg" in self.keyfilter:
            self.bg = {}
            for i, cam in enumerate(self.cameras):
                try:
                    imagepath = "experiments/dryice1/data/cam{}/bg.jpg".format(cam)
                    image = np.asarray(Image.open(imagepath), dtype=np.uint8).transpose((2, 0, 1)).astype(np.float32)
                    self.bg[cam] = image
                except:
                    pass