Dataloader add jpeg support?
crapthings opened this issue · 0 comments
crapthings commented
i get this error
assert len(self.images) == len(self.gts)
It seems that if the folder contains both .jpg and .jpeg file types, it can lead to errors.
Line 38 in 7863410
images = [os.path.join(image_root, f) for f in os.listdir(image_root) if f.lower().endswith(('.jpg', '.png'))]
images = sort(images)
gts = [os.path.join(gt_root, f) for f in os.listdir(gt_root) if f.lower().endswith(('.jpg', '.png'))]
gts = sort(gts)
self.images = [os.path.join(root, f) for f in os.listdir(root) if f.lower().endswith(('.jpg', '.png', '.jpeg'))]