Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1 about edge-informed-sisr

[richardburleigh]

It means at least one of your images isn't formatted correctly (eg. greyscale).

Here's a hacked together script I made to find and move those images to another folder. Just be aware that I haven't done any checks for filenames, so any other file in the directory will be moved as well.

import os
from scipy.misc import imread
from tqdm import tqdm

directory = 'INDIRECTORY'
movedir = 'OUTDIRECTORY'

for filename in tqdm(os.listdir(directory)):
        file = directory + filename

        try:
            im = imread(file)
            #print(filename + " is " + str(im.shape))
            if im.shape != (1024,1024,3):
                try:
                    os.rename(file, movedir + filename)
                except:
                    continue
                print(filename + " is " + str(im.shape))
        except:
            print("Failed to load " + str(filename))
            try:
                os.rename(file, movedir + filename)
            except:
                continue