fix bug of yolov8detect

2024-12-03 16:42:36 +08:00
parent 7338c5c06e
commit 3d4a3526a9
1 changed files with 1 additions and 1 deletions
@@ -71,7 +71,7 @@ class YoloV8Detect:
                else:
                    ret_yolo_masks.append(torch.zeros((1, _image.size[1], _image.size[0]), dtype=torch.float32))
                    # ret_yolo_masks.append(image2mask(Image.new('L', _image.size, "black")))
-                    log(f"{NODE_NAME} mask or box not detected.")
+                    log(f"{self.NODE_NAME} mask or box not detected.")

                # merge mask
                _mask = ret_yolo_masks[0]