Unverified Commit a83bb45f authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix repo consistency

parent 243439a8
...@@ -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})
......
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