栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

cv2无法存储图片到固定文件路径

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

cv2无法存储图片到固定文件路径

项目场景:

开源人脸识别项目,新增功能,实现把陌生的人脸存储在本地的固定目录下。

pytorch、win10、CUDA11.4


问题描述、

存储陌生人脸的过程中,cv2.imwrite不存储图片,无报错信息。
相关代码如下:

for k in range(len(faces)):
    # -------------------------------保存未注册人脸和已注册人脸并更新人脸库------------------------
    if results[k].item() == -1:
        if len(path_file_number) != 0:
            person_name = 'unknown_{}'.format(len(path_file_number) + count)
        else:
            person_name = 'unknown_{}'.format(count)

        save_path = self.save_path / person_name
        if not save_path.exists():
            save_path.mkdir()
        # cv2.imwrite(f'{save_path}/{person_name}_{datetime.datetime.now()}.jpg', faces[k].cpu().numpy())  #
        cv2.imwrite(
            faces[k].cpu().numpy())  # 保存人脸
        count += 1
        self.targets, self.names = update_facebank(self.conf, self.recognizer.model, self.detector, tta=self.tta)                            f"{save_path}/{person_name}_{datetime.datetime.strftime(time, '%Y-%m-%d %H:%M:%S')}.jpg",

    else:
        person_name = self.names[results[k] + 1]
        # save_path = Path(os.path.dirname(os.path.realpath(__file__)), 'data/result')
        # save_path = save_path / person_name
        save_path = self.save_path / person_name  #
        if not save_path.exists():
            save_path.mkdir()
        cv2.imwrite(
            f"{save_path}/{person_name}_{datetime.datetime.strftime(time, '%Y-%m-%d %H:%M:%S')}.jpg",
            faces[k].cpu().numpy())  # 保存人脸

原因分析:

cv2.imwrite() 方法的问题,路径写法在linux系统下可以正常运行,windows下路径无法正常识别。


解决方案:

把代码中的相对地址,替换成成绝对地址:

f"{save_path}/{person_name}_{datetime.datetime.strftime(time, '%Y-%m-%d %H:%M:%S')}.jpg"
filename = 'unknow_1.jpg'
file_path = os.path.join('InsightFace/models/data/facebank/',filename)
cv2.imwrite(file_path,faces[0].cpu().numpy())  # 保存人脸

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/768452.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号