def batch()

in eval/writers/videowriter.py [0:0]

• 21 lines of code
• 7 McCabe index (conditional complexity)

    def batch(self, iternum, itemnum, irgbrec, ialpharec=None, image=None, irgbsqerr=None, **kwargs):
        irgbrec = irgbrec.data.to("cpu").numpy().transpose((0, 2, 3, 1))
        if ialpharec is not None:
            ialpharec = ialpharec.data.to("cpu").numpy()[:, 0, :, :, None]
        else:
            ialpharec = 1.0

        # color correction
        imgout = irgbrec * self.colcorrect[None, None, None, :]

        # composite background color
        imgout = imgout + (1. - ialpharec) * self.bgcolor[None, None, None, :]

        # concatenate ground truth image
        if self.showtarget and image is not None:
            image = image.data.to("cpu").numpy().transpose((0, 2, 3, 1))
            image = image * self.colcorrect[None, None, None, :]
            imgout = np.concatenate((imgout, image), axis=2)

        # concatenate difference image
        if self.showdiff and imagediff is not None:
            irgbsqerr = np.mean(irgbsqerr.data.to("cpu").numpy(), axis=1)
            irgbsqerr = (cm.magma(4. * irgbsqerr / 255.)[:, :, :, :3] * 255.)
            imgout = np.concatenate((imgout, irgbsqerr), axis=2)

        self.writepool.map(writeimage,
                zip([self.randid for i in range(itemnum.size(0))],
                    itemnum.data.to("cpu").numpy(),
                    imgout))
        self.nitems += itemnum.size(0)