代码拉取完成,页面将自动刷新
同步操作将从 新无止竞/Ultra-Light-Fast-Generic-Face-Detector-1MB 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
#!/usr/bin/ python3
# -*- coding: utf-8 -*-
# @Time : 2019-10-17
# @Author : vealocia
# @FileName: evaluation_on_widerface.py
import math
import os
import sys
import cv2
from vision.ssd.config.fd_config import define_img_size
input_img_size = 320 # define input size ,default optional(128/160/320/480/640/1280)
define_img_size(input_img_size) # must put define_img_size() before 'import create_mb_tiny_fd, create_mb_tiny_fd_predictor'
from vision.ssd.mb_tiny_fd import create_mb_tiny_fd, create_mb_tiny_fd_predictor
from vision.ssd.mb_tiny_RFB_fd import create_Mb_Tiny_RFB_fd, create_Mb_Tiny_RFB_fd_predictor
label_path = "./models/voc-model-labels.txt"
# net_type = "mb_tiny_fd" # inference faster,lower precision
net_type = "mb_tiny_RFB_fd" # inference lower,higher precision
class_names = [name.strip() for name in open(label_path).readlines()]
num_classes = len(class_names)
# test_device = "cuda:0"
test_device = "cpu"
candidate_size = 500
threshold = 0.11
val_image_root = "/Users/mui/Desktop/coding/WIDER_val/images/" # path to widerface valuation image root
val_result_txt_save_root = "./widerface_evaluation/" # result directory
if net_type == 'mb_tiny_fd':
model_path = "models/pretrained/Mb_Tiny_FD_train_input_320.pth"
net = create_mb_tiny_fd(len(class_names), is_test=True, device=test_device)
predictor = create_mb_tiny_fd_predictor(net, candidate_size=candidate_size, device=test_device)
elif net_type == 'mb_tiny_RFB_fd':
model_path = "models/pretrained/Mb_Tiny_RFB_FD_train_input_320.pth"
net = create_Mb_Tiny_RFB_fd(len(class_names), is_test=True, device=test_device)
predictor = create_Mb_Tiny_RFB_fd_predictor(net, candidate_size=candidate_size, device=test_device)
else:
print("The net type is wrong!")
sys.exit(1)
net.load(model_path)
counter = 0
for parent, dir_names, file_names in os.walk(val_image_root):
for file_name in file_names:
if not file_name.lower().endswith('jpg'):
continue
im = cv2.imread(os.path.join(parent, file_name), cv2.IMREAD_COLOR)
im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
boxes, labels, probs = predictor.predict(im, candidate_size / 2, threshold)
event_name = parent.split('/')[-1]
if not os.path.exists(os.path.join(val_result_txt_save_root, event_name)):
os.makedirs(os.path.join(val_result_txt_save_root, event_name))
fout = open(os.path.join(val_result_txt_save_root, event_name, file_name.split('.')[0] + '.txt'), 'w')
fout.write(file_name.split('.')[0] + '\n')
fout.write(str(boxes.size(0)) + '\n')
for i in range(boxes.size(0)):
bbox = boxes[i, :]
fout.write('%d %d %d %d %.03f' % (math.floor(bbox[0]), math.floor(bbox[1]), math.ceil(bbox[2] - bbox[0]), math.ceil(bbox[3] - bbox[1]), probs[i] if probs[i] <= 1 else 1) + '\n')
fout.close()
counter += 1
print('[%d] %s is processed.' % (counter, file_name))
# note: with score_threshold = 0.11 and hard_nms, MAP of 320-input model on widerface val set is: 0.785/0.695/0.431
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。