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
dlib
Commits
a2c152e8
Commit
a2c152e8
authored
Sep 18, 2016
by
Davis King
Browse files
Made --cluster skip empty images
parent
b040e654
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
2 deletions
+16
-2
tools/imglab/src/cluster.cpp
tools/imglab/src/cluster.cpp
+16
-2
No files found.
tools/imglab/src/cluster.cpp
View file @
a2c152e8
...
@@ -111,6 +111,18 @@ double mean_aspect_ratio (
...
@@ -111,6 +111,18 @@ double mean_aspect_ratio (
// ----------------------------------------------------------------------------------------
// ----------------------------------------------------------------------------------------
bool
has_non_ignored_boxes
(
const
image_dataset_metadata
::
image
&
img
)
{
for
(
auto
&&
b
:
img
.
boxes
)
{
if
(
!
b
.
ignore
)
return
true
;
}
return
false
;
}
// ----------------------------------------------------------------------------------------
int
cluster_dataset
(
int
cluster_dataset
(
const
dlib
::
command_line_parser
&
parser
const
dlib
::
command_line_parser
&
parser
)
)
...
@@ -140,8 +152,9 @@ int cluster_dataset(
...
@@ -140,8 +152,9 @@ int cluster_dataset(
for
(
unsigned
long
i
=
0
;
i
<
data
.
images
.
size
();
++
i
)
for
(
unsigned
long
i
=
0
;
i
<
data
.
images
.
size
();
++
i
)
{
{
pbar
.
print_status
(
i
);
pbar
.
print_status
(
i
);
if
(
data
.
images
[
i
].
boxes
.
size
()
==
0
)
if
(
!
has_non_ignored_boxes
(
data
.
images
[
i
])
)
continue
;
continue
;
array2d
<
rgb_pixel
>
img
,
chip
;
array2d
<
rgb_pixel
>
img
,
chip
;
load_image
(
img
,
data
.
images
[
i
].
filename
);
load_image
(
img
,
data
.
images
[
i
].
filename
);
...
@@ -181,8 +194,9 @@ int cluster_dataset(
...
@@ -181,8 +194,9 @@ int cluster_dataset(
{
{
idata
[
i
].
first
=
std
::
numeric_limits
<
double
>::
infinity
();
idata
[
i
].
first
=
std
::
numeric_limits
<
double
>::
infinity
();
idata
[
i
].
second
.
filename
=
data
.
images
[
i
].
filename
;
idata
[
i
].
second
.
filename
=
data
.
images
[
i
].
filename
;
if
(
data
.
images
[
i
].
boxes
.
size
()
==
0
)
if
(
!
has_non_ignored_boxes
(
data
.
images
[
i
])
)
continue
;
continue
;
for
(
unsigned
long
j
=
0
;
j
<
data
.
images
[
i
].
boxes
.
size
();
++
j
)
for
(
unsigned
long
j
=
0
;
j
<
data
.
images
[
i
].
boxes
.
size
();
++
j
)
{
{
idata
[
i
].
second
.
boxes
.
push_back
(
data
.
images
[
i
].
boxes
[
j
]);
idata
[
i
].
second
.
boxes
.
push_back
(
data
.
images
[
i
].
boxes
[
j
]);
...
...
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