Skip to content

Commit 7253e81

Browse files
wangg12fmassa
authored andcommitted
change aspect ratio calculation formula (#1194)
1 parent c5533a2 commit 7253e81

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

references/detection/group_by_aspect_ratio.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def __len__(self):
104104
for _i, (img, _) in enumerate(data_loader):
105105
pbar.update(1)
106106
height, width = img.shape[-2:]
107-
aspect_ratio = float(height) / float(width)
107+
aspect_ratio = float(width) / float(height)
108108
aspect_ratios.append(aspect_ratio)
109109
return aspect_ratios
110110

@@ -115,7 +115,7 @@ def _compute_aspect_ratios_custom_dataset(dataset, indices=None):
115115
aspect_ratios = []
116116
for i in indices:
117117
height, width = dataset.get_height_and_width(i)
118-
aspect_ratio = float(height) / float(width)
118+
aspect_ratio = float(width) / float(height)
119119
aspect_ratios.append(aspect_ratio)
120120
return aspect_ratios
121121

@@ -126,7 +126,7 @@ def _compute_aspect_ratios_coco_dataset(dataset, indices=None):
126126
aspect_ratios = []
127127
for i in indices:
128128
img_info = dataset.coco.imgs[dataset.ids[i]]
129-
aspect_ratio = float(img_info["height"]) / float(img_info["width"])
129+
aspect_ratio = float(img_info["width"]) / float(img_info["height"])
130130
aspect_ratios.append(aspect_ratio)
131131
return aspect_ratios
132132

@@ -138,7 +138,7 @@ def _compute_aspect_ratios_voc_dataset(dataset, indices=None):
138138
for i in indices:
139139
# this doesn't load the data into memory, because PIL loads it lazily
140140
width, height = Image.open(dataset.images[i]).size
141-
aspect_ratio = float(height) / float(width)
141+
aspect_ratio = float(width) / float(height)
142142
aspect_ratios.append(aspect_ratio)
143143
return aspect_ratios
144144

0 commit comments

Comments
 (0)