我使用的版本:
pytorch 1.1.0
tensorboardX 1.7
easydict 1.9(这个用pip装,用conda装会把pytorch删掉)
python 3.6.6(之前是3.7,easydict好像只兼容py2.7和3.6,这个有待验证,为避免麻烦,直接换成3.6)
对代码的改动:1.导入的包的路径问题(路径最好写全);2.作者把文件夹名称和py文件命名一样,导致报错,一定要明确写出是哪个函数(不要命名相同);3.程序中有些相对路径存在问题;4.py2转py3。
VGG+shanghaitech_part_B
里面的代码有用到 *,如:
standard_transforms.Normalize(*mean_std)
* 就是去括号用的,mean_std可能是(),[ ],加上* 后 mean_std的括号就去掉了。
用tensorboard显示,运行如下代码:
tensorboard --logdir=C:\Users\Administrator\Desktop\exp\06-19_00-16_SHHB_VGG_1e-05
=后面要写DESKTOP-AC2IEHE文件具体的路径。训练结果如下: