Update segment_mapper.py

modelscope · Aug 27, 2024 · ae5dcc4 · ae5dcc4
1 parent ee372c3
commit ae5dcc4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/data_juicer/ops/mapper/segment_mapper.py b/data_juicer/ops/mapper/segment_mapper.py
@@ -21,8 +21,7 @@
 @OPERATORS.register_module(OP_NAME)
 @LOADED_IMAGES.register_module(OP_NAME)
 class SegmentMapper(Mapper):
-    """Perform segment-anything on images and 
-    return the bounding box values"""
+    """Perform segment-anything on images and return the bounding boxes."""
 
     _accelerator = 'cuda'
     _batched_op = True