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
chenpangpang
transformers
Commits
a83bb45f
Unverified
Commit
a83bb45f
authored
Oct 31, 2022
by
Sylvain Gugger
Browse files
Fix repo consistency
parent
243439a8
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
14 additions
and
13 deletions
+14
-13
src/transformers/models/conditional_detr/feature_extraction_conditional_detr.py
...s/conditional_detr/feature_extraction_conditional_detr.py
+14
-13
No files found.
src/transformers/models/conditional_detr/feature_extraction_conditional_detr.py
View file @
a83bb45f
...
@@ -1044,12 +1044,13 @@ class ConditionalDetrFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtrac
...
@@ -1044,12 +1044,13 @@ class ConditionalDetrFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtrac
# Get segmentation map and segment information of batch item
# Get segmentation map and segment information of batch item
target_size
=
target_sizes
[
i
]
if
target_sizes
is
not
None
else
None
target_size
=
target_sizes
[
i
]
if
target_sizes
is
not
None
else
None
segmentation
,
segments
=
compute_segments
(
segmentation
,
segments
=
compute_segments
(
mask_probs_item
,
mask_probs
=
mask_probs_item
,
pred_scores_item
,
pred_scores
=
pred_scores_item
,
pred_labels_item
,
pred_labels
=
pred_labels_item
,
mask_threshold
,
mask_threshold
=
mask_threshold
,
overlap_mask_area_threshold
,
overlap_mask_area_threshold
=
overlap_mask_area_threshold
,
target_size
,
label_ids_to_fuse
=
[],
target_size
=
target_size
,
)
)
# Return segmentation map in run-length encoding (RLE) format
# Return segmentation map in run-length encoding (RLE) format
...
@@ -1138,13 +1139,13 @@ class ConditionalDetrFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtrac
...
@@ -1138,13 +1139,13 @@ class ConditionalDetrFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtrac
# Get segmentation map and segment information of batch item
# Get segmentation map and segment information of batch item
target_size
=
target_sizes
[
i
]
if
target_sizes
is
not
None
else
None
target_size
=
target_sizes
[
i
]
if
target_sizes
is
not
None
else
None
segmentation
,
segments
=
compute_segments
(
segmentation
,
segments
=
compute_segments
(
mask_probs_item
,
mask_probs
=
mask_probs_item
,
pred_scores_item
,
pred_scores
=
pred_scores_item
,
pred_labels_item
,
pred_labels
=
pred_labels_item
,
mask_threshold
,
mask_threshold
=
mask_threshold
,
overlap_mask_area_threshold
,
overlap_mask_area_threshold
=
overlap_mask_area_threshold
,
label_ids_to_fuse
,
label_ids_to_fuse
=
label_ids_to_fuse
,
target_size
,
target_size
=
target_size
,
)
)
results
.
append
({
"segmentation"
:
segmentation
,
"segments_info"
:
segments
})
results
.
append
({
"segmentation"
:
segmentation
,
"segments_info"
:
segments
})
...
...
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