2922|2

83

帖子

1

TA的资源

一粒金砂(中级)

楼主
 

人脸识别模型建立过程 [复制链接]

  本帖最后由 swzswz 于 2022-10-21 19:08 编辑

 

人脸识别模型

 

人脸识别模型步骤

人脸识别主要分为两个模型:

  • 检测网模型

  • 分类网模型

模型建立过程

视频录制转换

按照海思开发文档的store_sample例程,修改对应参数。将视频录制出来,或者利用手机进行视频录制。

编译完后,利用NFS将板端的视频复制到电脑本地上。

搭建FFmpeg环境,将视频裁剪成图片。

数据集制作和标注

利用labelme工具,标注数据集。

 

 

模型训练

利用CUDA进行云端模型训练,我们用的海思服务器。

 

模型转换

一般是用pytorch-caffe转换。

 

模型量化

通过caffe模型转换成板端推理的wk模型。

模型部署板端和调试

 

视频演示

​​

1

代码以及其他文件

 static HI_S32 CnnTrashClassifyFlag(const RecogNumInfo items[], HI_S32 itemNum, HI_CHAR* buf, HI_S32 size)
 {
     HI_S32 offset = 0;
     HI_CHAR *trashName = NULL;
 ​
     offset += snprintf_s(buf + offset, size - offset, size - offset - 1, "face_classify: {");
     for (HI_U32 i = 0; i < itemNum; i++) {
         const RecogNumInfo *item = &items;
         uint32_t score = item->score * HI_PER_BASE / SCORE_MAX;
         if (score < THRESH_MIN) {
             break;
         }
         SAMPLE_PRT("----trash item flag----num:%d, score:%d\n", item->num, score);
         switch (item->num) {
             case 0u:
                 trashName = "face";
                 UartSendRead(uartFd1,face);
                 SAMPLE_PRT("----trash name----:%s\n", trashName);
                 break;
             case 1u:
                 trashName = "background";
                 UartSendRead(uartFd1,background);
                 SAMPLE_PRT("----trash name----:%s\n", trashName);
                 break;
             default:
                 trashName = "unknown";
                 UartSendRead(uartFd1,unknown);
                 SAMPLE_PRT("----trash name----:%s\n", trashName);
                 break;
         }
         offset += snprintf_s(buf + offset, size - offset, size - offset - 1,
             "%s%s %u:%u%%", (i == 0 ? " " : ", "), trashName, (int)item->num, (int)score);
         HI_ASSERT(offset < size);
     }
     offset += snprintf_s(buf + offset, size - offset, size - offset - 1, " }");
     HI_ASSERT(offset < size);
     return HI_SUCCESS;
 } 

 

code.zip

15.02 KB, 下载次数: 0

ip_camera.py

892 Bytes, 下载次数: 0

txt.py

212 Bytes, 下载次数: 0

face_classcify_inst.wk

11.04 MB, 下载次数: 0

2022年嵌入式大赛海思赛道开发指导.pdf

44.86 MB, 下载次数: 0

ohos_camera_ai_demo

122.23 KB, 下载次数: 0

ohos_store_demo

40.77 KB, 下载次数: 0

ohos_vio_demo

70.55 KB, 下载次数: 0

sample_ai.conf

303 Bytes, 下载次数: 0

最新回复

看起来很高大上的,感谢分享!   详情 回复 发表于 2022-10-21 22:26
点赞 关注
 
 

回复
举报

7059

帖子

11

TA的资源

版主

沙发
 

看起来很高大上的,感谢分享!

点评

用海思开发套件弄的,效果还不错!  详情 回复 发表于 2022-11-10 16:45
 
 
 

回复

83

帖子

1

TA的资源

一粒金砂(中级)

板凳
 
lugl4313820 发表于 2022-10-21 22:26 看起来很高大上的,感谢分享!

用海思开发套件弄的,效果还不错!

 
 
 

回复
您需要登录后才可以回帖 登录 | 注册

随便看看
查找数据手册?

EEWorld Datasheet 技术支持

相关文章 更多>>
关闭
站长推荐上一条 1/6 下一条

 
EEWorld订阅号

 
EEWorld服务号

 
汽车开发圈

About Us 关于我们 客户服务 联系方式 器件索引 网站地图 最新更新 手机版

站点相关: 国产芯 安防电子 汽车电子 手机便携 工业控制 家用电子 医疗电子 测试测量 网络通信 物联网

北京市海淀区中关村大街18号B座15层1530室 电话:(010)82350740 邮编:100190

电子工程世界版权所有 京B2-20211791 京ICP备10001474号-1 电信业务审批[2006]字第258号函 京公网安备 11010802033920号 Copyright © 2005-2025 EEWORLD.com.cn, Inc. All rights reserved
快速回复 返回顶部 返回列表