search

ShuweiShao / AF-SfMLearner

[MedIA2022 & ICRA2021] Self-Supervised Monocular Depth and Ego-Motion Estimation in Endoscopy: Appearance Flow to the Rescue
MIT License
101 stars 15 forks source link

非常感谢作者公开您的工作 #19

Closed 1158191255 closed 1 year ago

1158191255 commented 1 year ago

顺便提一下，如果使用高版本torchvision，需要将mono_dataset.py中的color_avg=transforms.ColorJitter.get_params(self.brightness,self.contrast,self.saturation,self.hue)改成color_avg=transforms.ColorJitter(self.brightness,self.contrast,self.saturation,self.hue) 因为get_params这个函数在高版本里发生了修改。

ShuweiShao commented 1 year ago

谢谢您的提醒，我将会更新在Important Note中。