in data/dryice1.py [0:0]
def get_background(self, bg):
if "bg" in self.keyfilter:
for i, cam in enumerate(self.cameras):
if cam in self.bg:
bg[cam].data[:] = torch.from_numpy(self.bg[cam]).to("cuda")