Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +1 -0
my_model/KBVQA.py
CHANGED
@@ -134,6 +134,7 @@ class KBVQA:
|
|
134 |
"""
|
135 |
|
136 |
image = self.detector.process_image(img)
|
|
|
137 |
detected_objects_string, detected_objects_list = self.detector.detect_objects(image, threshold=self.detection_confidence)
|
138 |
image_with_boxes = self.detector.draw_boxes(img, detected_objects_list)
|
139 |
return image_with_boxes, detected_objects_string
|
|
|
134 |
"""
|
135 |
|
136 |
image = self.detector.process_image(img)
|
137 |
+
st.write("thresh",self.detection_confidence, self.detection_model)
|
138 |
detected_objects_string, detected_objects_list = self.detector.detect_objects(image, threshold=self.detection_confidence)
|
139 |
image_with_boxes = self.detector.draw_boxes(img, detected_objects_list)
|
140 |
return image_with_boxes, detected_objects_string
|