segmentation-backend

Sleeping

diegokauer commited on Dec 27, 2023

Commit

49221d6

1 Parent(s): c38e7cf

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -85,7 +85,7 @@ class Model(LabelStudioMLBase):
                         'width': x2-x,
                         'height': y2-y,
                         'rotation': 0,
-                        'labels': [self.id2label[label]]
                     }
                 })

                         'width': x2-x,
                         'height': y2-y,
                         'rotation': 0,
+                        'labels': [self.id2label[label.item()]]
                     }
                 })