def process_3d(self, data_dir):
new_data_dir = '{}_new'.format(data_dir.rstrip('/'))
if os.path.exists(new_data_dir):
shutil.rmtree(new_data_dir)
os.makedirs(new_data_dir)
for filename in FileHelper.list_dir(data_dir):
if not ImageHelper.is_img(filename) or 'depth' in filename:
Log.info('Image Path: {}'.format(os.path.join(data_dir, filename)))
continue
file_path = os.path.join(data_dir, filename)
img = io.imread(file_path)
kpts = self.detect_face(img)
if kpts is None:
Log.info('Invliad face detected in {}'.format(file_path))
continue
depth = np.array(io.imread(os.path.join(data_dir, filename.replace('rgb', 'depth'))))
face_depth, kpts = self.align_face([np.array(img), np.array(depth)], kpts)
if face_depth is None:
Log.info('Invliad face detected in {}'.format(file_path))
continue
ImageHelper.save(ImageHelper.rgb2bgr(face_depth[0]), os.path.join(new_data_dir, filename))
ImageHelper.save(ImageHelper.rgb2bgr(face_depth[1]), os.path.join(new_data_dir, filename.replace('rgb', 'depth')))
def vis_bboxes(self,
image_in,
bboxes_list,
name='default',
sub_dir='bbox'):
"""
Show the diff bbox of individuals.
"""
base_dir = os.path.join(self.configer.get('project_dir'), DET_DIR,
sub_dir)
if isinstance(image_in, Image.Image):
image = ImageHelper.rgb2bgr(ImageHelper.to_np(image_in))
else:
image = image_in.copy()
if not os.path.exists(base_dir):
log.error('Dir:{} not exists!'.format(base_dir))
os.makedirs(base_dir)
img_path = os.path.join(
base_dir,
name if ImageHelper.is_img(name) else '{}.jpg'.format(name))
for bbox in bboxes_list:
image = cv2.rectangle(image, (bbox[0], bbox[1]),
(bbox[2], bbox[3]), (0, 255, 0), 2)
cv2.imwrite(img_path, image)
def process(self, data_dir):
new_data_dir = '{}_new'.format(data_dir.rstrip('/'))
if os.path.exists(new_data_dir):
shutil.rmtree(new_data_dir)
os.makedirs(new_data_dir)
for filename in FileHelper.list_dir(data_dir):
if not ImageHelper.is_img(filename):
Log.info('Image Path: {}'.format(os.path.join(data_dir, filename)))
continue
file_path = os.path.join(data_dir, filename)
img = io.imread(file_path)
kpts = self.detect_face(img)
if kpts is None:
Log.info('Invliad face detected in {}'.format(file_path))
continue
face, kpts = self.align_face(img, kpts)
cv2.imwrite(os.path.join(new_data_dir, filename), ImageHelper.rgb2bgr(face))