diff --git a/scripts/analyze_temp_features.py b/scripts/analyze_temp_features.py index 712db90..ca52d7f 100644 --- a/scripts/analyze_temp_features.py +++ b/scripts/analyze_temp_features.py @@ -64,8 +64,8 @@ def analyze_validation_set( # 加载灰度图像(用于温度提取) gray_img = load_infrared_image(str(img_path)) - # 进行检测 - detections = detector.detect(str(img_path), conf=conf) + # 进行检测(复用已加载的图像) + detections = detector.detect(gray_img, conf=conf) for det in detections: bbox = det['bbox'] diff --git a/scripts/usage_example.py b/scripts/usage_example.py index 9bc135a..c3c2abb 100644 --- a/scripts/usage_example.py +++ b/scripts/usage_example.py @@ -54,8 +54,8 @@ def example3_temperature_extraction(): # 读取图像 gray_img = cv2.imread(SAMPLE_IMAGE, cv2.IMREAD_GRAYSCALE) - # 检测目标 - detections = detector.detect(SAMPLE_IMAGE) + # 检测目标(复用已加载的图像) + detections = detector.detect(gray_img) # 提取第一个检测框的温度特征 if detections: diff --git a/src/detect.py b/src/detect.py index 0b6e403..0458de2 100644 --- a/src/detect.py +++ b/src/detect.py @@ -92,7 +92,7 @@ class InfraredDetector: else: img = image.copy() - detections = self.detect(image, conf) + detections = self.detect(img, conf) for det in detections: x1, y1, x2, y2 = map(int, det['bbox'])