Commit 0bb31540 authored by Filip Lux's avatar Filip Lux
Browse files

fix mask flist

parent bfec8760

src/dataset.py

+1 −1
Original line number Diff line number Diff line
@@ -437,7 +437,7 @@ class Dataset: 
                if self.seq_length > 0:

                    imgs = [imgs[i % len(imgs)] for i in range(self.seq_length)]

                    gts = [gts[i % len(gts)] for i in range(self.seq_length)]

                    masks = [masks[i % len(masks)] for i in range(self.seq_length)]

                    # masks = [masks[i % len(masks)] for i in range(self.seq_length)]



                img_flist.extend(imgs)

                gt_flist.extend(gts)