Update detectYOLOv8.m

matlab-deep-learning · Feb 28, 2024 · a74bcb6 · a74bcb6
1 parent 509bd62
commit a74bcb6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/detectYOLOv8.m b/detectYOLOv8.m
@@ -43,4 +43,8 @@
 [bboxes,scores,labelIds] = helper.postprocess(outFeatureMaps, ...
      origSize, newSize, numClasses);
 
+bboxes = gather(bboxes);
+scores = gather(scores);
+labelIds = gather(labelIds);
+
 end