Function isimage

src/finetune_data.py:110–112

(path)

def isimage(path):
    if 'png' in path.lower() or 'jpg' in path.lower() or 'jpeg' in path.lower():
        return True


class MaskBase(Dataset):