3dc20bc15b767ad60edfc15154b630dbef8b79cc,onmt/IO.py,,read_img_file,#Any#Any#Any#Any#,94

Before Change

    
    assert not truncate, "truncate not implemented yet!"
    with codecs.open(path, "r", "utf-8") as corpus_file:
        for i, line in enumerate(corpus_file):
            img_path = os.path.join(src_img_dir, line.strip())
            if not os.path.exists(img_path):
                img_path = line

After Change

            assert os.path.exists(img_path), \
                "img path %s not found" % (line.strip())
            img = transforms.ToTensor()(Image.open(img_path))
            if truncate and truncate != (0, 0):
                if not (img.size(1) <= truncate[0]
                        and img.size(2) <= truncate[1]):
                    continue
            example_dict = {side: img,
                            side+"_path": line.strip(),
                            "indices": index}
            index += 1

Instances