def get_pts_for_mirror_image(file_path, verbose=False):
for image_file_name in mio.image_paths(file_path):
if os.path.exists(
os.path.splitext(str(image_file_name))[0] +
'_mirror.jpg') and not os.path.exists(
os.path.splitext(str(image_file_name))[0] + '_mirror.pts'):
img = mio.import_image(image_file_name).mirror()
mirrored_points = []
for i in range(68):
unmirrored_points = img.landmarks.get('PTS').lms.points
mirrored_points.append([
unmirrored_points[mirror_transfrom[i + 1] - 1][1],
unmirrored_points[mirror_transfrom[i + 1] - 1][0]
])
if verbose:
print('Adding mirror pts for: ' + image_file_name)
with open(
os.path.splitext(str(image_file_name))[0] + '_mirror.pts',
'w') as f:
f.write('version: 1\nn_points: %d\n{\n' %
(len(mirrored_points)))
for point in mirrored_points:
f.write('%f %f\n' % (point[0], point[1]))
f.write('}')
def from_menpo(cls, data_dir, verbose=True):
"""
Create class instance from directory of menpo files
Parameters
----------
data_dir: string
path to data directory
verbose: bool
whether or not to print current progress
Returns
-------
class instance
"""
if verbose:
print("Loading data from %s" % data_dir)
wrapper_fn = tqdm
else:
def linear_wrapper(x):
return x
wrapper_fn = linear_wrapper
img_paths = list(mio.image_paths(data_dir))
samples = []
for _img in wrapper_fn(img_paths):
samples.append(
SingleImage.from_menpo(mio.import_image(_img), img_file=_img))
return cls(samples=samples)
def resolve_all_paths(img_paths_or_patterns):
img_paths = set()
for img_path_or_pattern in img_paths_or_patterns:
if not isfile(img_path_or_pattern):
img_paths.update(set(mio.image_paths(img_path_or_pattern)))
else:
img_paths.add(Path(img_path_or_pattern))
return img_paths
def test_image_paths():
ls = mio.image_paths(mio.data_dir_path())
assert(len(list(ls)) == 6)
def test_image_paths():
ls = mio.image_paths(mio.data_dir_path())
assert (len(list(ls)) == 6)
def kf_sequence_image_paths(i, expression):
return LazyList.init_from_iterable(
mio.image_paths(kf_sequence_path(i, expression)))
def test_image_paths():
ls = mio.image_paths(os.path.join(mio.data_dir_path(), '*'))
assert(len(ls) == 5)
def test_image_paths():
ls = mio.image_paths(os.path.join(mio.data_dir_path(), '*'))
assert(len(list(ls)) == 6)
