环境配置完之后运行训练代码遇到了这个问题:
产生问题的原因可能是python版本的原因,解决办法:
将 mono_dataset.py文件中的
transforms.ColorJitter.get_params(
self.brightness, self.contrast, self.saturation, self.hue)
改为
transforms.ColorJitter(
self.brightness, self.contrast, self.saturation, self.hue)
在mono_dataset.py中包含两个这样的问题,分别替换问题得到解决