Toybrick

标题: MTCNN例程复现问题 [打印本页]

作者: yaowei 时间: 2019-6-18 10:36
标题: MTCNN例程复现问题
教程给出的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的人脸检测出来了一部分，如下图所示

[attach]342[/attach][attach]343[/attach]

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

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

Traceback (most recent call last):

  File "demo.py", line 554, in <module>

    main()

  File "demo.py", line 533, in main

    boundingboxes, points = detect_face(img_matlab, minsize, pnet_rknn_list, rnet_rknn, onet_rknn, threshold, False, factor)

  File "demo.py", line 283, in detect_face

    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])

ValueError: cannot reshape array of size 25740 into shape (1,2,91,69)
复制代码

还看研究为什么会导致这样的问题。25740是更高分辨率的检测结果。

作者: 395876134 时间: 2019-6-19 15:00
你好，请参照http://t.rock-chips.com/forum.ph ... amp;highlight=mtcnn这边的说明。

作者: yaowei 时间: 2019-6-26 09:52
说明里面有哪里可以参考吗？可能是转换精度的影响吗？

作者: 395876134 时间: 2019-6-27 08:24
你可以看下我们官方rknn相关的文档，里头有说明精度的问题。

作者: puyanan 时间: 2019-7-4 11:18
本帖最后由 puyanan 于 2019-7-4 11:26 编辑

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

作者: yaowei 时间: 2019-7-4 11:35
按理来说是这样的，但是我没有继续研究，放弃了这条线。

欢迎光临 Toybrick (https://t.rock-chips.com/)