optimize comment

Author: Mark-ZhouWX

research/segment-anything/segment_anything/__init__.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 from .build_sam import (
     build_sam,
     build_sam_vit_h,

research/segment-anything/segment_anything/automatic_mask_generator.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 import numpy as np
 import mindspore as ms
 from mindspore import ops

research/segment-anything/segment_anything/build_sam.py

@@ -1,8 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
 import os.path
 
 import mindspore as ms

research/segment-anything/segment_anything/evaluate/evaluator.py

@@ -2,7 +2,7 @@
 
 import mindspore as ms
 from mindspore import nn
-from mindspore.dataset import RepeatDataset
+from mindspore.dataset import BatchDataset
 from tqdm import tqdm
 
 from segment_anything.utils import logger
@@ -15,7 +15,7 @@ class Evaluator:
     """
     def __init__(self,
                  network: nn.Cell,
-                 data_loader: RepeatDataset,
+                 data_loader: BatchDataset,
                  metric: List[ms.train.Metric],
                  input_column: List[List[str]],
                  ):

research/segment-anything/segment_anything/evaluate/metrics.py

@@ -40,7 +40,6 @@ def eval(self):
         num_step = all_reduce(self.num_step)
         miou = sum_iou / num_step
         print(sum_iou, num_step, miou)
-        # miou = self.sum_iou / self.num_step
         miou = miou.asnumpy()
         res_dict = dict(miou=miou)
         return res_dict

research/segment-anything/segment_anything/modeling/__init__.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 from .sam import Sam
 from .image_encoder import ImageEncoderViT
 from .mask_decoder import MaskDecoder

research/segment-anything/segment_anything/modeling/common.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 import mindspore as ms
 from mindspore import nn, ops
 

research/segment-anything/segment_anything/modeling/image_encoder.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 import mindspore as ms
 from mindspore import nn, ops
 

research/segment-anything/segment_anything/modeling/mask_decoder.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 import mindspore as ms
 from mindspore import nn, ops
 

research/segment-anything/segment_anything/modeling/prompt_encoder.py

@@ -1,9 +1,3 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-
 import numpy as np
 import mindspore as ms
 from mindspore import nn, ops