(img_id, detail, out_dir)
| 18 | |
| 19 | |
| 20 | def generate_labels(img_id, detail, out_dir): |
| 21 | |
| 22 | def _class_to_index(mask, _mapping, _key): |
| 23 | # assert the values |
| 24 | values = np.unique(mask) |
| 25 | for i in range(len(values)): |
| 26 | assert (values[i] in _mapping) |
| 27 | index = np.digitize(mask.ravel(), _mapping, right=True) |
| 28 | return _key[index].reshape(mask.shape) |
| 29 | |
| 30 | mask = Image.fromarray( |
| 31 | _class_to_index(detail.getMask(img_id), _mapping=_mapping, _key=_key)) |
| 32 | filename = img_id['file_name'] |
| 33 | mask.save(osp.join(out_dir, filename.replace('jpg', 'png'))) |
| 34 | return osp.splitext(osp.basename(filename))[0] |
| 35 | |
| 36 | |
| 37 | def parse_args(): |
nothing calls this directly
no test coverage detected