Toybrick

MTCNN例程复现问题

yaowei

中级会员

积分
369
发表于 2019-6-18 10:36:54    查看: 2077|回复: 5 | [复制链接]    打印 | 显示全部楼层
教程给出的MTCNN RKNN demo(https://github.com/chenshiqin/mtcnn),复现出现一些问题,如下:

1. 环境:PC Ubuntu;RKNN 1.0.0,使用原demo中转换的rknn权重文件,没有重新转换。

问题:运行demo.py,出现的问题是检测结果与例程中不一样,test1.jpg检测结果很好,两张人脸都检测出来;test2.jpg的人脸没有检测出来;test3和test4的人脸检测出来了一部分,如下图所示



运行demo_camera.py,用的是笔记本自带的摄像头,人脸检测效果不好,有时候可以检测出来,有时候检测不出来,而且运行速度很慢。

2. 环境:PC Ubuntu,RKNN 1.0.0,用例程中自带的程序进行权重转换,运行demo.py 和demo_camera.py,出现以下错误:

  1. Traceback (most recent call last):
  2.   File "demo.py", line 554, in <module>
  3.     main()
  4.   File "demo.py", line 533, in main
  5.     boundingboxes, points = detect_face(img_matlab, minsize, pnet_rknn_list, rnet_rknn, onet_rknn, threshold, False, factor)
  6.   File "demo.py", line 283, in detect_face
  7.     out_prob1=out_prob1.reshape(PNET_PYRAMID_ARR[i][0][0],PNET_PYRAMID_ARR[i][0][1], PNET_PYRAMID_ARR[i][0][2], PNET_PYRAMID_ARR[i][0][3])
  8. ValueError: cannot reshape array of size 25740 into shape (1,2,91,69)
复制代码
还看研究为什么会导致这样的问题。25740是更高分辨率的检测结果。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

395876134

中级会员

积分
480
发表于 2019-6-19 15:00:14 | 显示全部楼层
你好,请参照http://t.rock-chips.com/forum.ph ... amp;highlight=mtcnn这边的说明。
回复

使用道具 举报

yaowei

中级会员

积分
369
 楼主| 发表于 2019-6-26 09:52:59 | 显示全部楼层
说明里面有哪里可以参考吗?可能是转换精度的影响吗?
回复

使用道具 举报

395876134

中级会员

积分
480
发表于 2019-6-27 08:24:32 | 显示全部楼层
你可以看下我们官方rknn相关的文档,里头有说明精度的问题。
回复

使用道具 举报

puyanan

中级会员

积分
206
发表于 2019-7-4 11:18:23 | 显示全部楼层
本帖最后由 puyanan 于 2019-7-4 11:26 编辑

demo.py中有两个参数PNET_PYRAMID, PNET_PYRAMID_ARR,这个是和输入图片的尺寸对应的。如果测试图片尺寸不是450x344x3的话,就要改写这两个参数。而且要重新做模型转换。报错的地方是否是这里出了问题?
回复

使用道具 举报

yaowei

中级会员

积分
369
 楼主| 发表于 2019-7-4 11:35:56 | 显示全部楼层
按理来说是这样的,但是我没有继续研究,放弃了这条线。
回复

使用道具 举报

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

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表