update v1.4.1

2025-04-02 19:27:43 +08:00
parent 467652bd69
commit 6c8af8d7a8
31 changed files with 2134 additions and 58 deletions
@@ -111,9 +111,11 @@ def pack_imagelist_into_tensor(image_list):
        image_tensor.append(img.view(c, h * w).transpose(1, 0))  # h*w, c
        shapes.append((h, w))

-    image_tensor = pad_sequence(image_tensor, batch_first=True).permute(0, 2, 1)  # b, c, l
+    image_tensor = pad_sequence(image_tensor,
+                                batch_first=True).permute(0, 2, 1)  # b, c, l
    return image_tensor, shapes

+
 def limit_batch_data(batch_data_list, log_num):
    if log_num and log_num > 0:
        batch_data_list_limited = []
@@ -123,4 +125,4 @@ def limit_batch_data(batch_data_list, log_num):
            batch_data_list_limited.append(sub_data)
        return batch_data_list_limited
    else:
-        return batch_data_list
+        return batch_data_list