Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1#!/usr/bin/env python
2# coding=utf-8
5"""Data loading code"""
8import os
9import functools
11import PIL.Image
13from .sample import DelayedSample
16def load_pil(path):
17 """Loads a sample data
19 Parameters
20 ----------
22 path : str
23 The full path leading to the image to be loaded
26 Returns
27 -------
29 image : PIL.Image.Image
30 A PIL image
32 """
34 return PIL.Image.open(path)
37def load_pil_baw(path):
38 """Loads a sample data
40 Parameters
41 ----------
43 path : str
44 The full path leading to the image to be loaded
47 Returns
48 -------
50 image : PIL.Image.Image
51 A PIL image in grayscale mode
53 """
55 return load_pil(path).convert("L")
58def load_pil_rgb(path):
59 """Loads a sample data
61 Parameters
62 ----------
64 path : str
65 The full path leading to the image to be loaded
68 Returns
69 -------
71 image : PIL.Image.Image
72 A PIL image in RGB mode
74 """
76 return load_pil(path).convert("RGB")
79def make_delayed(sample, loader, key=None):
80 """Returns a delayed-loading Sample object
82 Parameters
83 ----------
85 sample : dict
86 A dictionary that maps field names to sample data values (e.g. paths)
88 loader : object
89 A function that inputs ``sample`` dictionaries and returns the loaded
90 data.
92 key : str
93 A unique key identifier for this sample. If not provided, assumes
94 ``sample`` is a dictionary with a ``data`` entry and uses its path as
95 key.
98 Returns
99 -------
101 sample : bob.med.tb.data.sample.DelayedSample
102 In which ``key`` is as provided and ``data`` can be accessed to trigger
103 sample loading.
105 """
107 return DelayedSample(
108 functools.partial(loader, sample),
109 key=key or sample["data"],
110 label=sample["label"]
111 )