Skip to content

Commit 029d69b

Browse files
committed
undo no-op diffs
1 parent 6211846 commit 029d69b

File tree

46 files changed

+302
-252
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

46 files changed

+302
-252
lines changed

inference_models/inference_models/models/deep_lab_v3_plus/deep_lab_v3_plus_segmentation_onnx.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -311,9 +311,10 @@ def post_process(
311311
conf = result.confidence.clone()
312312
seg_map[below] = self._background_class_id
313313
conf[below] = 0.0
314-
result = SemanticSegmentationResult(
315-
segmentation_map=seg_map,
316-
confidence=conf,
314+
results.append(
315+
SemanticSegmentationResult(
316+
segmentation_map=seg_map,
317+
confidence=conf,
318+
)
317319
)
318-
results.append(result)
319320
return results

inference_models/inference_models/models/deep_lab_v3_plus/deep_lab_v3_plus_segmentation_torch.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -286,9 +286,10 @@ def post_process(
286286
conf = result.confidence.clone()
287287
seg_map[below] = self._background_class_id
288288
conf[below] = 0.0
289-
result = SemanticSegmentationResult(
290-
segmentation_map=seg_map,
291-
confidence=conf,
289+
results.append(
290+
SemanticSegmentationResult(
291+
segmentation_map=seg_map,
292+
confidence=conf,
293+
)
292294
)
293-
results.append(result)
294295
return results

inference_models/inference_models/models/deep_lab_v3_plus/deep_lab_v3_plus_segmentation_trt.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -376,11 +376,12 @@ def post_process(
376376
conf = result.confidence.clone()
377377
seg_map[below] = self._background_class_id
378378
conf[below] = 0.0
379-
result = SemanticSegmentationResult(
380-
segmentation_map=seg_map,
381-
confidence=conf,
379+
results.append(
380+
SemanticSegmentationResult(
381+
segmentation_map=seg_map,
382+
confidence=conf,
383+
)
382384
)
383-
results.append(result)
384385
self._post_process_stream.synchronize()
385386
return results
386387

inference_models/inference_models/models/dinov3/dinov3_classification_onnx.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -380,9 +380,10 @@ def post_process(
380380
predicted_classes = torch.argwhere(
381381
batch_element_confidence >= thresholds
382382
).squeeze(dim=-1)
383-
prediction = MultiLabelClassificationPrediction(
384-
class_ids=predicted_classes,
385-
confidence=batch_element_confidence,
383+
results.append(
384+
MultiLabelClassificationPrediction(
385+
class_ids=predicted_classes,
386+
confidence=batch_element_confidence,
387+
)
386388
)
387-
results.append(prediction)
388389
return results

inference_models/inference_models/models/dinov3/dinov3_classification_torch.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -358,9 +358,10 @@ def post_process(
358358
predicted_classes = torch.argwhere(
359359
batch_element_confidence >= thresholds
360360
).squeeze(dim=-1)
361-
prediction = MultiLabelClassificationPrediction(
362-
class_ids=predicted_classes,
363-
confidence=batch_element_confidence,
361+
results.append(
362+
MultiLabelClassificationPrediction(
363+
class_ids=predicted_classes,
364+
confidence=batch_element_confidence,
365+
)
364366
)
365-
results.append(prediction)
366367
return results

inference_models/inference_models/models/resnet/resnet_classification_onnx.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -364,9 +364,10 @@ def post_process(
364364
predicted_classes = torch.argwhere(
365365
batch_element_confidence >= thresholds
366366
).squeeze(dim=-1)
367-
prediction = MultiLabelClassificationPrediction(
368-
class_ids=predicted_classes,
369-
confidence=batch_element_confidence,
367+
results.append(
368+
MultiLabelClassificationPrediction(
369+
class_ids=predicted_classes,
370+
confidence=batch_element_confidence,
371+
)
370372
)
371-
results.append(prediction)
372373
return results

inference_models/inference_models/models/resnet/resnet_classification_torch.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -342,9 +342,10 @@ def post_process(
342342
predicted_classes = torch.argwhere(
343343
batch_element_confidence >= thresholds
344344
).squeeze(dim=-1)
345-
prediction = MultiLabelClassificationPrediction(
346-
class_ids=predicted_classes,
347-
confidence=batch_element_confidence,
345+
results.append(
346+
MultiLabelClassificationPrediction(
347+
class_ids=predicted_classes,
348+
confidence=batch_element_confidence,
349+
)
348350
)
349-
results.append(prediction)
350351
return results

inference_models/inference_models/models/resnet/resnet_classification_trt.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -474,11 +474,12 @@ def post_process(
474474
predicted_classes = torch.argwhere(
475475
batch_element_confidence >= thresholds
476476
).squeeze(dim=-1)
477-
prediction = MultiLabelClassificationPrediction(
478-
class_ids=predicted_classes,
479-
confidence=batch_element_confidence,
477+
results.append(
478+
MultiLabelClassificationPrediction(
479+
class_ids=predicted_classes,
480+
confidence=batch_element_confidence,
481+
)
480482
)
481-
results.append(prediction)
482483
self._post_process_stream.synchronize()
483484
return results
484485

inference_models/inference_models/models/rfdetr/rfdetr_object_detection_onnx.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -275,10 +275,11 @@ def post_process(
275275
image_detections=selected_boxes_xyxy,
276276
image_metadata=image_meta,
277277
)
278-
detections = Detections(
279-
xyxy=selected_boxes_xyxy.round().int(),
280-
confidence=predicted_confidence,
281-
class_id=top_classes.int(),
278+
results.append(
279+
Detections(
280+
xyxy=selected_boxes_xyxy.round().int(),
281+
confidence=predicted_confidence,
282+
class_id=top_classes.int(),
283+
)
282284
)
283-
results.append(detections)
284285
return results

inference_models/inference_models/models/rfdetr/rfdetr_object_detection_pytorch.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -533,10 +533,11 @@ def post_process(
533533
device=boxes.device,
534534
)
535535
boxes[:, :4].add_(static_crop_offsets)
536-
detections = Detections(
537-
xyxy=boxes.round().int(),
538-
confidence=scores,
539-
class_id=labels.int(),
536+
detections_list.append(
537+
Detections(
538+
xyxy=boxes.round().int(),
539+
confidence=scores,
540+
class_id=labels.int(),
541+
)
540542
)
541-
detections_list.append(detections)
542543
return detections_list

0 commit comments

Comments
 (0)