def __init__(self):
self.face_detector = FaceDetector()
self.face_landmark = FaceLandmark()
self.trace = GroupTrack()
###another thread should run detector in a slow way and update the track_box
self.track_box=None
self.previous_image=None
self.previous_box=None
self.diff_thres=5
self.top_k = cfg.DETECT.topk
self.iou_thres=cfg.TRACE.iou_thres
self.alpha=cfg.TRACE.smooth_box
def image_demo(data_dir):
args.model
detector = FaceDetector(args.model)
count = 0
pics = []
GetFileList(data_dir, pics)
pics = [x for x in pics if 'jpg' in x or 'png' in x]
#pics.sort()
for pic in pics:
img = cv2.imread(pic)
img_show = img.copy()
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
star = time.time()
boxes = detector(img, 0.3)
#print('one iamge cost %f s'%(time.time()-star))
#print(boxes.shape)
#print(boxes)
################toxml or json
print(boxes.shape[0])
if boxes.shape[0] == 0:
print(pic)
for box_index in range(boxes.shape[0]):
bbox = boxes[box_index]
cv2.rectangle(img_show, (int(bbox[0]), int(bbox[1])),
(int(bbox[2]), int(bbox[3])), (255, 0, 0), 4)
# cv2.putText(img_show, str(bbox[4]), (int(bbox[0]), int(bbox[1]) + 30),
# cv2.FONT_HERSHEY_SIMPLEX, 1,
# (255, 0, 255), 2)
#
# cv2.putText(img_show, str(int(bbox[5])), (int(bbox[0]), int(bbox[1]) + 40),
# cv2.FONT_HERSHEY_SIMPLEX, 1,
# (0, 0, 255), 2)
cv2.namedWindow('res', 0)
cv2.imshow('res', img_show)
cv2.waitKey(0)
print(count)
def __init__(self):
self.face_detector = FaceDetector()
self.face_landmark = FaceLandmark()
self.trace = GroupTrack()
###another thread should run detector in a slow way and update the track_box
self.track_box = None
self.previous_image = None
self.previous_box = None
self.diff_thres = 5
self.top_k = cfg.DETECT.topk
self.min_face = cfg.DETECT.min_face
self.iou_thres = cfg.TRACE.iou_thres
self.alpha = cfg.TRACE.smooth_box
if 'ema' in cfg.TRACE.ema_or_one_euro:
self.filter = EmaFilter(self.alpha)
else:
self.filter = OneEuroFilter()
ap.add_argument( "--model", required=True, default='', help="model to eval:")
ap.add_argument( "--is_show", required=False, default=False, help="show result or not?")
ap.add_argument( "--data_dir", required=False, default="./FDDB/img", help="dir to img")
ap.add_argument( "--split_dir", required=False,default='./FDDB/FDDB-folds',help="dir to FDDB-folds")
ap.add_argument( "--result", required=False,default='./result',help="dir to write result")
args = ap.parse_args()
IMAGES_DIR = args.data_dir
ANNOTATIONS_PATH = args.split_dir
RESULT_DIR = args.result
if not os.access(RESULT_DIR,os.F_OK):
os.mkdir(RESULT_DIR)
face_detector = FaceDetector(args.model)
annotations = [s for s in os.listdir(ANNOTATIONS_PATH) if s.endswith('ellipseList.txt')]
image_lists = [s for s in os.listdir(ANNOTATIONS_PATH) if not s.endswith('ellipseList.txt')]
annotations = sorted(annotations)
image_lists = sorted(image_lists)
images_to_use = []
for n in image_lists:
with open(os.path.join(ANNOTATIONS_PATH, n)) as f:
images_to_use.extend(f.readlines())
images_to_use = [s.strip() for s in images_to_use]
with open(os.path.join(RESULT_DIR, 'faceList.txt'), 'w') as f:
for p in images_to_use:
f.write(p + '\n')
ap.add_argument("--split_dir",
required=False,
default='../FDDB/FDDB-folds',
help="dir to FDDB-folds")
ap.add_argument("--result",
required=False,
default='./resultFDDB',
help="dir to write result")
args = ap.parse_args()
IMAGES_DIR = args.data_dir
ANNOTATIONS_PATH = args.split_dir
RESULT_DIR = args.result
MODEL_PATH = args.model
face_detector = FaceDetector([MODEL_PATH])
annotations = [
s for s in os.listdir(ANNOTATIONS_PATH) if s.endswith('ellipseList.txt')
]
image_lists = [
s for s in os.listdir(ANNOTATIONS_PATH)
if not s.endswith('ellipseList.txt')
]
annotations = sorted(annotations)
image_lists = sorted(image_lists)
images_to_use = []
for n in image_lists:
with open(os.path.join(ANNOTATIONS_PATH, n)) as f:
images_to_use.extend(f.readlines())
max_score = np.max(det_accu[:, 4])
det_accu_sum = np.zeros((1, 5))
det_accu_sum[:, 0:4] = np.sum(det_accu[:, 0:4], axis=0) / np.sum(
det_accu[:, -1:])
det_accu_sum[:, 4] = max_score
try:
dets = np.row_stack((dets, det_accu_sum))
except:
dets = det_accu_sum
dets = dets[0:750, :]
return dets
#face_detector = FaceDetector([MODEL_META_PATH,MODEL_PATH])
face_detector = FaceDetector(['./model/detector.pb'])
predictions = []
for n in tqdm(images_to_use):
image_array = cv2.imread(os.path.join(IMAGES_DIR, n) + '.jpg')
image_array = cv2.cvtColor(image_array, cv2.COLOR_BGR2RGB)
# threshold is important to set low
boxes = face_detector(image_array, score_threshold=0.05)
# ##flip det but seems no needs for fddb
# flip_img=np.flip(image_array,1)
#
# boxes_flip_ = face_detector(flip_img, score_threshold=0.05)
# boxes_flip = np.zeros(boxes_flip_.shape)
# boxes_flip[:, 0] = flip_img.shape[1] - boxes_flip_[:, 2]
# boxes_flip[:, 1] = boxes_flip_[:, 1]
ap.add_argument("--split_dir",
required=False,
default='./FDDB/FDDB-folds',
help="dir to FDDB-folds")
ap.add_argument("--result",
required=False,
default='./result',
help="dir to write result")
args = ap.parse_args()
IMAGES_DIR = args.data_dir
ANNOTATIONS_PATH = args.split_dir
RESULT_DIR = args.result
MODEL_PATH = args.model
INPUT_SHAPE = (args.input_shape, args.input_shape)
face_detector = FaceDetector(MODEL_PATH)
annotations = [
s for s in os.listdir(ANNOTATIONS_PATH) if s.endswith('ellipseList.txt')
]
image_lists = [
s for s in os.listdir(ANNOTATIONS_PATH)
if not s.endswith('ellipseList.txt')
]
annotations = sorted(annotations)
image_lists = sorted(image_lists)
images_to_use = []
for n in image_lists:
with open(os.path.join(ANNOTATIONS_PATH, n)) as f:
images_to_use.extend(f.readlines())
import glob
import cv2
import os
import time
import matplotlib.pyplot as plt
import tensorflow as tf
from lib.core.api.face_detector import FaceDetector
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
detector = FaceDetector(['./model/detector_2.pb'])
def GetFileList(dir, fileList):
newDir = dir
if os.path.isfile(dir):
fileList.append(dir)
elif os.path.isdir(dir):
for s in os.listdir(dir):
# if s == "pts":
# continue
newDir = os.path.join(dir, s)
GetFileList(newDir, fileList)
return fileList
def facedetect():
count = 0
# print('t')
# data_dir = '/Users/andreyrizhiy/face_detect/ASF/*.jpg'
# pics = []