(self, img, crop_size=None)
| 257 | return score |
| 258 | |
| 259 | def process_image(self, img, crop_size=None): |
| 260 | p_img = img |
| 261 | |
| 262 | if img.shape[2] < 3: |
| 263 | im_b = p_img |
| 264 | im_g = p_img |
| 265 | im_r = p_img |
| 266 | p_img = np.concatenate((im_b, im_g, im_r), axis=2) |
| 267 | |
| 268 | p_img = normalize(p_img, self.image_mean, self.image_std) |
| 269 | |
| 270 | if crop_size is not None: |
| 271 | p_img, margin = pad_image_to_shape(p_img, crop_size, cv2.BORDER_CONSTANT, value=0) |
| 272 | p_img = p_img.transpose(2, 0, 1) |
| 273 | |
| 274 | return p_img, margin |
| 275 | |
| 276 | p_img = p_img.transpose(2, 0, 1) |
| 277 | |
| 278 | return p_img |
no test coverage detected