From 8090d633e9eaa04398f13b9c5b30ef1327e6b7a5 Mon Sep 17 00:00:00 2001
From: sunyanfang01 <sunyanfang01@baidu.com>
Date: Fri, 5 Jun 2020 16:44:45 +0800
Subject: [PATCH] add blazeface

---
 paddlex/cv/datasets/widerface.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddlex/cv/datasets/widerface.py b/paddlex/cv/datasets/widerface.py
index 70cd148..45ea796 100644
--- a/paddlex/cv/datasets/widerface.py
+++ b/paddlex/cv/datasets/widerface.py
@@ -156,6 +156,7 @@ class WIDERFACEDetection(VOCDetection):
                         self.file_list.append([img_file, voc_rec])
                         im_ct += 1
                         
+        self.num_samples = len(self.file_list)
         self.coco_gt = COCO()
         self.coco_gt.dataset = annotations
         self.coco_gt.createIndex()
-- 
GitLab