add wider_val.txt file

Author: clibdev

.gitignore

@@ -1,3 +1,5 @@
 */__pycache__
 /.idea
 *.pt
+/data/widerface/val/images
+/widerface_evaluate/widerface_txt

README.md

@@ -2,8 +2,9 @@
 
 Differences between original repository and fork:
 
-* Compatibility with PyTorch >=2.0
+* Compatibility with PyTorch >=2.0. (🔥)
 * Installation with [requirements.txt](requirements.txt) file.
+* The wider_val.txt file for WIDERFace evaluation. 
 * No deprecations. The following deprecations has been fixed:
   * UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument.
 

data/widerface/val/wider_val.txt

@@ -112,7 +112,7 @@ def detect(model, img0):
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('--weights', nargs='+', type=str, default='runs/train/exp5/weights/last.pt', help='model.pt path(s)')
+    parser.add_argument('--weights', nargs='+', type=str, default='weights/yolov5s-face.pt', help='model.pt path(s)')
     parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
     parser.add_argument('--conf-thres', type=float, default=0.02, help='object confidence threshold')
     parser.add_argument('--iou-thres', type=float, default=0.5, help='IOU threshold for NMS')
@@ -125,8 +125,8 @@ def detect(model, img0):
     parser.add_argument('--name', default='exp', help='save results to project/name')
     parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
     parser.add_argument('--save_folder', default='./widerface_evaluate/widerface_txt/', type=str, help='Dir to save txt results')
-    parser.add_argument('--dataset_folder', default='../WiderFace/val/images/', type=str, help='dataset path')
-    parser.add_argument('--folder_pict', default='/yolov5-face/data/widerface/val/wider_val.txt', type=str, help='folder_pict')
+    parser.add_argument('--dataset_folder', default='./data/widerface/val/images/', type=str, help='dataset path')
+    parser.add_argument('--folder_pict', default='./data/widerface/val/wider_val.txt', type=str, help='folder_pict')
     opt = parser.parse_args()
     print(opt)
 
@@ -145,7 +145,7 @@ def detect(model, img0):
         # testing dataset
         testset_folder = opt.dataset_folder
 
-        for image_path in tqdm(glob.glob(os.path.join(testset_folder, '*'))):
+        for image_path in tqdm(glob.glob(os.path.join(testset_folder, '**/*'))):
             if image_path.endswith('.txt'):
                 continue
             img0 = cv2.imread(image_path)  # BGR