Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
7253e816
Commit
7253e816
authored
Aug 06, 2019
by
Gu Wang
Committed by
Francisco Massa
Aug 05, 2019
Browse files
change aspect ratio calculation formula (#1194)
parent
c5533a2c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
references/detection/group_by_aspect_ratio.py
references/detection/group_by_aspect_ratio.py
+4
-4
No files found.
references/detection/group_by_aspect_ratio.py
View file @
7253e816
...
@@ -104,7 +104,7 @@ def _compute_aspect_ratios_slow(dataset, indices=None):
...
@@ -104,7 +104,7 @@ def _compute_aspect_ratios_slow(dataset, indices=None):
for
_i
,
(
img
,
_
)
in
enumerate
(
data_loader
):
for
_i
,
(
img
,
_
)
in
enumerate
(
data_loader
):
pbar
.
update
(
1
)
pbar
.
update
(
1
)
height
,
width
=
img
.
shape
[
-
2
:]
height
,
width
=
img
.
shape
[
-
2
:]
aspect_ratio
=
float
(
height
)
/
float
(
width
)
aspect_ratio
=
float
(
width
)
/
float
(
height
)
aspect_ratios
.
append
(
aspect_ratio
)
aspect_ratios
.
append
(
aspect_ratio
)
return
aspect_ratios
return
aspect_ratios
...
@@ -115,7 +115,7 @@ def _compute_aspect_ratios_custom_dataset(dataset, indices=None):
...
@@ -115,7 +115,7 @@ def _compute_aspect_ratios_custom_dataset(dataset, indices=None):
aspect_ratios
=
[]
aspect_ratios
=
[]
for
i
in
indices
:
for
i
in
indices
:
height
,
width
=
dataset
.
get_height_and_width
(
i
)
height
,
width
=
dataset
.
get_height_and_width
(
i
)
aspect_ratio
=
float
(
height
)
/
float
(
width
)
aspect_ratio
=
float
(
width
)
/
float
(
height
)
aspect_ratios
.
append
(
aspect_ratio
)
aspect_ratios
.
append
(
aspect_ratio
)
return
aspect_ratios
return
aspect_ratios
...
@@ -126,7 +126,7 @@ def _compute_aspect_ratios_coco_dataset(dataset, indices=None):
...
@@ -126,7 +126,7 @@ def _compute_aspect_ratios_coco_dataset(dataset, indices=None):
aspect_ratios
=
[]
aspect_ratios
=
[]
for
i
in
indices
:
for
i
in
indices
:
img_info
=
dataset
.
coco
.
imgs
[
dataset
.
ids
[
i
]]
img_info
=
dataset
.
coco
.
imgs
[
dataset
.
ids
[
i
]]
aspect_ratio
=
float
(
img_info
[
"
height
"
])
/
float
(
img_info
[
"
width
"
])
aspect_ratio
=
float
(
img_info
[
"
width
"
])
/
float
(
img_info
[
"
height
"
])
aspect_ratios
.
append
(
aspect_ratio
)
aspect_ratios
.
append
(
aspect_ratio
)
return
aspect_ratios
return
aspect_ratios
...
@@ -138,7 +138,7 @@ def _compute_aspect_ratios_voc_dataset(dataset, indices=None):
...
@@ -138,7 +138,7 @@ def _compute_aspect_ratios_voc_dataset(dataset, indices=None):
for
i
in
indices
:
for
i
in
indices
:
# this doesn't load the data into memory, because PIL loads it lazily
# this doesn't load the data into memory, because PIL loads it lazily
width
,
height
=
Image
.
open
(
dataset
.
images
[
i
]).
size
width
,
height
=
Image
.
open
(
dataset
.
images
[
i
]).
size
aspect_ratio
=
float
(
height
)
/
float
(
width
)
aspect_ratio
=
float
(
width
)
/
float
(
height
)
aspect_ratios
.
append
(
aspect_ratio
)
aspect_ratios
.
append
(
aspect_ratio
)
return
aspect_ratios
return
aspect_ratios
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment