search

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
284 stars 38 forks source link

RGB dataset #58

Open yug125lk opened 10 months ago

yug125lk commented 10 months ago

Hi, thank you again. I used the code to train my custom dataset pair dataset; (input image, target image). It works when I convert the dataset to grayscale, but it gave me this error with RGB:

RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [10, 2, 128, 128, 3]