-
- Downloads
Modified the data_folder to be more generic
Loading
-
@onikisins : This should work for both the cases; I am not sure
torch.Tensor(img_array_transformed).unsqueeze(0)
is needed here. Can you check if this is ok with you. -
Thanks @ageorge, I will try to run an experiment with this modification, to make sure there are no side-effects. Will let you know once done.
Please register or sign in to comment