Jelajahi Sumber

fix bugs in meter reader case'

FlyingQianMM 4 tahun lalu
induk
melakukan
5e836e7357

+ 1 - 1
examples/meter_reader/deploy/python/reader_deploy.py

@@ -214,7 +214,7 @@ class MeterReader:
             im_size = min(meter_num, i + seg_batch_size)
             meter_images = list()
             for j in range(i, im_size):
-                meter_images.append(resized_meters[j - i])
+                meter_images.append(resized_meters[j])
             # Segment scales and point in each meter area
             result = self.segmenter.batch_predict(
                 transforms=self.seg_transforms, img_file_list=meter_images)

+ 1 - 1
examples/meter_reader/reader_infer.py

@@ -214,7 +214,7 @@ class MeterReader:
             im_size = min(meter_num, i + seg_batch_size)
             meter_images = list()
             for j in range(i, im_size):
-                meter_images.append(resized_meters[j - i])
+                meter_images.append(resized_meters[j])
             # Segment scales and point in each meter area
             result = self.segmenter.batch_predict(
                 transforms=self.seg_transforms, img_file_list=meter_images)