Open YuhsiHu opened 2 years ago
Did you solve this problem, I have this problem too
No, I didn't.
I guess this may be caused by different pytorch versions. The transforms.ColorJitter.get_params function in datasets.py returns a tuple, and the tuple cannot be called as a function in the transforms.Compose function. I tried a working method, modifying datasets.py:
fn_idx, brightness_factor, contrast_factor, saturation_factor, hue_factor = transforms.ColorJitter.get_params( brightness=brightness, contrast=contrast, saturation=saturation, hue=hue ) color_trans = transforms.Compose([ transforms.ToPILImage(), transforms.ColorJitter(brightness=brightness_factor, contrast=contrast_factor, saturation=saturation_factor), ]) Hope this helps you.
Thank you for sharing this great paper and excellent code. I followed the README file in CVA-MVSNet. When I tried to train the model, there was a TypeError. Could you please help me? Thank you for your time.