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:
@@ -64,8 +64,8 @@ def analyze_validation_set(
|
|||||||
# 加载灰度图像(用于温度提取)
|
# 加载灰度图像(用于温度提取)
|
||||||
gray_img = load_infrared_image(str(img_path))
|
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:
|
for det in detections:
|
||||||
bbox = det['bbox']
|
bbox = det['bbox']
|
||||||
|
|||||||
@@ -54,8 +54,8 @@ def example3_temperature_extraction():
|
|||||||
# 读取图像
|
# 读取图像
|
||||||
gray_img = cv2.imread(SAMPLE_IMAGE, cv2.IMREAD_GRAYSCALE)
|
gray_img = cv2.imread(SAMPLE_IMAGE, cv2.IMREAD_GRAYSCALE)
|
||||||
|
|
||||||
# 检测目标
|
# 检测目标(复用已加载的图像)
|
||||||
detections = detector.detect(SAMPLE_IMAGE)
|
detections = detector.detect(gray_img)
|
||||||
|
|
||||||
# 提取第一个检测框的温度特征
|
# 提取第一个检测框的温度特征
|
||||||
if detections:
|
if detections:
|
||||||
|
|||||||
+1
-1
@@ -92,7 +92,7 @@ class InfraredDetector:
|
|||||||
else:
|
else:
|
||||||
img = image.copy()
|
img = image.copy()
|
||||||
|
|
||||||
detections = self.detect(image, conf)
|
detections = self.detect(img, conf)
|
||||||
|
|
||||||
for det in detections:
|
for det in detections:
|
||||||
x1, y1, x2, y2 = map(int, det['bbox'])
|
x1, y1, x2, y2 = map(int, det['bbox'])
|
||||||
|
|||||||
Reference in New Issue
Block a user