runing test.py encounter AssertionError

[LeviArckman]

what's wrong?

[wdcww]

In the example in the readme, the number of gt images must be equal to the number of mask images.

If you feel this is too troublesome, you can modify the logic in guided_diffusion/image_datasets.py.
For example:
I added the following code to this line, so that many gt images use the only mask image

# If there is only one mask image, reuse it
    if len(mask_paths) == 1 and len(gt_paths) > 1:
        mask_paths = [mask_paths[0]] * len(gt_paths)