Get mask from Camera_id, Frame_id Args: Camera_id (int/str of a number): Camera_id (str) in {'Camera_5mp': '0'~'47', 'Camera_12mp':'48'~'60', 'Kinect': '0'~'7'} Frame_id a.(int/str of a nu
(self, Camera_id, Frame_id=None,disable_tqdm=True)
| 184 | return cv2.imdecode(color_array, cv2.IMREAD_COLOR) |
| 185 | |
| 186 | def get_mask(self, Camera_id, Frame_id=None,disable_tqdm=True): |
| 187 | """Get mask from Camera_id, Frame_id |
| 188 | |
| 189 | Args: |
| 190 | Camera_id (int/str of a number): |
| 191 | Camera_id (str) in |
| 192 | {'Camera_5mp': '0'~'47', |
| 193 | 'Camera_12mp':'48'~'60', |
| 194 | 'Kinect': '0'~'7'} |
| 195 | Frame_id a.(int/str of a number): '0' ~ 'num_frame' |
| 196 | b.list of numbers (int/str) |
| 197 | c.None: get batch of all imgs in order of time sequence |
| 198 | Returns: |
| 199 | a single img : |
| 200 | 'color': HWC in bgr (uint8) |
| 201 | 'mask' : HW (uint8) |
| 202 | 'depth': HW (uint16) |
| 203 | """ |
| 204 | if not 'Mask' in self.smc: |
| 205 | print("=== no key: Mask.\nplease check available keys!") |
| 206 | return None |
| 207 | |
| 208 | Camera_id = str(Camera_id) |
| 209 | |
| 210 | assert(isinstance(Frame_id,(list,int, str, type(None)))) |
| 211 | if isinstance(Frame_id, (str,int)): |
| 212 | Frame_id = str(Frame_id) |
| 213 | assert(Frame_id in self.smc['Mask'][Camera_id]['mask'].keys()) |
| 214 | img_byte = self.smc['Mask'][Camera_id]['mask'][Frame_id][()] |
| 215 | img_color = self.__read_color_from_bytes__(img_byte) |
| 216 | img_color = np.max(img_color,2) |
| 217 | return img_color |
| 218 | else: |
| 219 | if Frame_id is None: |
| 220 | Frame_id_list =sorted([int(l) for l in self.smc['Mask'][Camera_id]['mask'].keys()]) |
| 221 | elif isinstance(Frame_id, list): |
| 222 | Frame_id_list = Frame_id |
| 223 | rs = [] |
| 224 | for fi in tqdm.tqdm(Frame_id_list, disable=disable_tqdm): |
| 225 | rs.append(self.get_mask(Camera_id,fi)) |
| 226 | return np.stack(rs,axis=0) |
| 227 | |
| 228 | def get_img(self, Camera_group, Camera_id, Image_type,Frame_id=None,disable_tqdm=True): |
| 229 | """Get image its Camera_group, Camera_id, Image_type and Frame_id |
no test coverage detected