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

fix generate_mini_batch bugs

FlyingQianMM преди 5 години
родител
ревизия
ed1d008960
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      paddlex/cv/datasets/dataset.py

+ 1 - 1
paddlex/cv/datasets/dataset.py

@@ -249,7 +249,7 @@ def generate_minibatch(batch_data, label_padding_value=255, mapper=None):
                 padding_batch.append((padding_im, ) + tuple(data[1:]))
         else:
             # padding the image of classification during the infering phase
-            padding_batch.append((padding_im))
+            padding_batch.append((padding_im, ))
     return padding_batch