Finding tiny face 论文 pytorch版

Finding tiny face 论文 pytorch版

1 准备代码

相关环境啥的自己配嗷。

开源代码下载链接https://github.com/varunagrawal/tiny-faces-pytorch

2 准备数据

在下载源码文件的data目录下创建相应数据文件，wider_face是论文使用的数据集，数据集需要自己下载http://shuoyang1213.me/WIDERFACE/index.html。

3 修改路径

此时就可以使用main.py文件进行训练了。

evaluate.py部分

先创建一个保存测试结果的文件夹results，下载权重文件weights，目录如图。

下面是我的参数设置

然后运行，运行结果奇奇怪怪，居然是个txt文件。

要是觉得运行时觉得文件太大，可以把wider_face_test_filelist.txt里的一些图片给删了。
删的时候要保证txt里的图片和wider里的图片相对应。

后面打算修改在原图像显示框框。。。。若是有大佬看出问题，望指正。觉得有用可以点个赞，哈哈。