Do we need batch here?

[zivzone]

Hi,
Thanks for your repo. I', wondering do we needs batch here?
for batch in range (batch_size):
I guess you just put one src image and one occ image as a pair into
color_transfer_sot

BR,
Ziv

[kennyvoo]

The batch side here refers to the solver batch instead of the image batch. This function supposes to take in 1 image pair as input. Although this function is taken from DeepFaceLab as mentioned in the code, It was originally from this source. You can refer to this website for more information about color transfer via sliced optimal transport.

[zivzone]

Hi, @kennyvoo
Ok, I see. Thanks for your reply!