()
| 58 | |
| 59 | |
| 60 | def get_inference_augmentor(): |
| 61 | # load ResNet mean from Kaiming: |
| 62 | # from tensorpack.utils.loadcaffe import get_caffe_pb |
| 63 | # obj = get_caffe_pb().BlobProto() |
| 64 | # obj.ParseFromString(open('ResNet_mean.binaryproto').read()) |
| 65 | # pp_mean_224 = np.array(obj.data).reshape(3, 224, 224).transpose(1,2,0) |
| 66 | |
| 67 | meta = ILSVRCMeta() |
| 68 | pp_mean = meta.get_per_pixel_mean() |
| 69 | pp_mean_224 = pp_mean[16:-16, 16:-16, :] |
| 70 | |
| 71 | transformers = [ |
| 72 | imgaug.ResizeShortestEdge(256), |
| 73 | imgaug.CenterCrop((224, 224)), |
| 74 | imgaug.MapImage(lambda x: x - pp_mean_224), |
| 75 | ] |
| 76 | return transformers |
| 77 | |
| 78 | |
| 79 | def run_test(params, input): |
no test coverage detected