Upload processor
Browse files- image_processing.py +5 -1
image_processing.py
CHANGED
|
@@ -34,7 +34,11 @@ class LSPDetrImageProcessor(BaseImageProcessorFast):
|
|
| 34 |
non_no_object_indices = labels != outputs["logits"].size(-1) - 1
|
| 35 |
|
| 36 |
return [
|
| 37 |
-
{
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
for b, indices in enumerate(non_no_object_indices)
|
| 39 |
]
|
| 40 |
|
|
|
|
| 34 |
non_no_object_indices = labels != outputs["logits"].size(-1) - 1
|
| 35 |
|
| 36 |
return [
|
| 37 |
+
{
|
| 38 |
+
"polygons": polygons[b, indices],
|
| 39 |
+
"labels": labels[b, indices],
|
| 40 |
+
"embeddings": outputs["embeddings"][b, indices],
|
| 41 |
+
}
|
| 42 |
for b, indices in enumerate(non_no_object_indices)
|
| 43 |
]
|
| 44 |
|