Skip to content

S
seminar-breakout
Commit 72d54fdc authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

augmentorlist in load-resnet

parent 0166e4a5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/ResNet/load-resnet.py
View file @ 72d54fdc
...@@ -94,7 +94,7 @@ def run_test(params, input): ...@@ -94,7 +94,7 @@ def run_test(params, input):
) )
predict_func = OfflinePredictor(pred_config) predict_func = OfflinePredictor(pred_config)
prepro = get_inference_augmentor() prepro = imgaug.AugmentorList(get_inference_augmentor())
im = cv2.imread(input).astype('float32') im = cv2.imread(input).astype('float32')
im = prepro.augment(im) im = prepro.augment(im)
im = np.reshape(im, (1, 224, 224, 3)) im = np.reshape(im, (1, 224, 224, 3))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment