Skip to content

Commit 0b5761d

Browse files
committed
rename variables
1 parent 0e7fe94 commit 0b5761d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

sbd.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -693,8 +693,8 @@ def predict_video(self, path, confidence_threshold=0.2, show_class_with_score=Tr
693693
break
694694
img, boxes = self.predict(self.model, img_bgr, device=self.primary_device, confidence_threshold=confidence_threshold, heatmap=heatmap)
695695
img = self.data_generator.resize(img, (view_width, view_height))
696-
boxed_image = self.draw_box(img, boxes, show_class_with_score=show_class_with_score)
697-
cv2.imshow('video', boxed_image)
696+
img = self.draw_box(img, boxes, show_class_with_score=show_class_with_score)
697+
cv2.imshow('video', img)
698698
key = cv2.waitKey(1)
699699
if key == 27:
700700
exit(0)
@@ -875,7 +875,7 @@ def training_view_function(self):
875875
img_path = np.random.choice(self.validation_image_paths)
876876
img, _ = self.data_generator.load_image(img_path)
877877
img, boxes = self.predict(self.model, img, device=self.primary_device, heatmap=True)
878-
boxed_image = self.draw_box(img, boxes)
878+
img = self.draw_box(img, boxes)
879879
cv2.imshow('training view', img)
880880
cv2.waitKey(1)
881881

0 commit comments

Comments
 (0)