OpenPCDet为KITTI数据集生成数据信息出现错误TypeError: load() missing 1 required positional argument: ‘Loader‘的解决方案

OpenPCDet为KITTI数据集生成数据信息出现错误

TypeError: load() missing 1 required positional argument: 'Loader'

通过查阅资料得知可能是作者的写法不是十分规范,所以我根据编写yaml.load()的方法,对pcdet/datasets/kitti/kitti_dataset.py文件中最后(大概在476行的位置)进行了修改。

作者原来的写法

 dataset_cfg = EasyDict(yaml.load(open(sys.argv[2])))

我修改后的代码

with open(sys.argv[2],"r") as f:
            dataset_cfg = EasyDict(yaml.load(f,Loader=yaml.FullLoader))

这里要注意缩进,是四个空格,建议不要使用tab键可能会出现问题。

修改后的代码可以生成数据集的数据信息,也就是多了以下几个文件和文件夹
其中imagesets,testing和training是原有的

参考文章:https://blog.csdn.net/liufuren620/article/details/103644413

posted @ 2022-05-10 09:46  Sunny不要停  阅读(483)  评论(0编辑  收藏  举报