fix: fix multiple image duplicate loading issues

- detect.py: detect_and_draw()中传给detect()的是已加载的img数组而非字符串路径
- usage_example.py: example3中复用gray_img数组而非重新加载
- analyze_temp_features.py: 复用已加载的gray_img数组进行检测
This commit is contained in:
SkyLi
2026-03-27 08:53:29 +08:00
parent 10fb804c14
commit 9a5f6fe473
3 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -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']
+2 -2
View File
@@ -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:
+1 -1
View File
@@ -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'])