Skip to content

S
seminar-breakout
Commit 8648d571 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

correct rounding in imagenet augmentor

parent 9bf59445
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/ResNet/imagenet_resnet_utils.py
View file @ 8648d571
...@@ -33,8 +33,8 @@ class GoogleNetResize(imgaug.ImageAugmentor): ...@@ -33,8 +33,8 @@ class GoogleNetResize(imgaug.ImageAugmentor):
for _ in range(10): for _ in range(10):
targetArea = self.rng.uniform(0.08, 1.0) * area targetArea = self.rng.uniform(0.08, 1.0) * area
aspectR = self.rng.uniform(0.75, 1.333) aspectR = self.rng.uniform(0.75, 1.333)
ww = int(np.sqrt(targetArea * aspectR)) ww = int(np.sqrt(targetArea * aspectR) + 0.5)
hh = int(np.sqrt(targetArea / aspectR)) hh = int(np.sqrt(targetArea / aspectR) + 0.5)
if self.rng.uniform() < 0.5: if self.rng.uniform() < 0.5:
ww, hh = hh, ww ww, hh = hh, ww
if hh <= h and ww <= w: if hh <= h and ww <= w:
......
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