Преглед на файлове

fix bugs in meter reader case'

FlyingQianMM преди 4 години
родител
ревизия
5e836e7357
променени са 2 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 1 1
      examples/meter_reader/deploy/python/reader_deploy.py
  2. 1 1
      examples/meter_reader/reader_infer.py

+ 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)