f22814f2405f83179c99f1fd4b895671bbdb124c,tools/scripts/features/lmdb_conversion.py,LMDBConversion,extract,#LMDBConversion#,76

Before Change

                    os.path.join(self.args.features_folder, file_base_name),
                    item["features"],
                )
                np.save(
                    os.path.join(self.args.features_folder, info_file_base_name),
                    tmp_dict,
                )

    def execute(self):
        if self.args.mode == "convert":
            self.convert()

After Change

                file_base_name = str(img_id) + ".npy"

                path = os.path.join(self.args.features_folder, file_base_name)
                if PathManager.exists(path):
                    continue
                info_path = os.path.join(self.args.features_folder, info_file_base_name)
                base_path = "/".join(path.split("/")[:-1])
                PathManager.mkdirs(base_path)
                np.save(PathManager.open(path, "wb"), item["features"])

Instances