(img_str)
| 274 | module.running_var = torch.ones_like(module.running_var) |
| 275 | |
| 276 | def pil_loader(img_str): |
| 277 | buff = io.BytesIO(img_str) |
| 278 | with Image.open(buff) as img: |
| 279 | img = img.convert('RGB') |
| 280 | return img |
| 281 | |
| 282 | def cv2_loader(img_str): |
| 283 | img_array = np.frombuffer(img_str, dtype=np.uint8) |
no test coverage detected