diff --git a/src/utils/__init__.py b/src/utils/__init__.py index 83ae01c..b8706a0 100644 --- a/src/utils/__init__.py +++ b/src/utils/__init__.py @@ -2,4 +2,5 @@ from .images_grid import ( create_images_grid_by_columns, create_images_grid_by_rows, ) +from grid_annotations import create_grid_annotations from .tensor_convert import tensor_to_pillow, pillow_to_tensor