Traceback (most recent call last):
File "train.py", line 544, in <module>
train(hyp, opt, device, tb_writer)
File "train.py", line 64, in train
data_dict = yaml.load(f, Loader=yaml.SafeLoader) # data dict
File "D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\yaml\__init__.py", line 79, in load
loader = Loader(stream)
File "D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\yaml\loader.py", line 34, in __init__
Reader.__init__(self, stream)
File "D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\yaml\reader.py", line 85, in __init__
self.determine_encoding()
File "D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\yaml\reader.py", line 124, in determine_encoding
self.update_raw()
File "D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\yaml\reader.py", line 178, in update_raw
data = self.stream.read(size)
UnicodeDecodeError: 'gbk' codec can't decode byte 0x80 in position 218: illegal multibyte sequence
翻译过来大致意思就是:gbk编码器不能解码位置在20的字节0x80:非法多字节序列。
通过报错的信息提示,我们找到代码所在的py文件:
找出错的位置,train.py---->train(hyp, opt, device, tb_writer)
定位成功
# Save run settings
with open(save_dir / 'hyp.yaml', 'w') as f:
yaml.dump(hyp, f, sort_keys=False)
with open(save_dir / 'opt.yaml', 'w') as f:
yaml.dump(vars(opt), f, sort_keys=False)
解决方法:
原因显而易见,with open打开文件的时候没有指定编码格式,默认使用了gbk,所以导致此次报错。
解决方法:给open()方法指定编码格式为utf-8:
# Save run settings
with open(save_dir / 'hyp.yaml', 'w', encoding='UTF-8') as f:
yaml.dump(hyp, f, sort_keys=False)
with open(save_dir / 'opt.yaml', 'w', encoding='UTF-8') as f:
yaml.dump(vars(opt), f, sort_keys=False)
注意:这个文件中的所有open()函数 都需要添加 encoding='UTF-8'