diff --git a/sam3/__init__.py b/sam3/__init__.py index 14270a6..1e75971 100644 --- a/sam3/__init__.py +++ b/sam3/__init__.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from .model_builder import build_sam3_image_model __version__ = "0.1.0" diff --git a/sam3/agent/__init__.py b/sam3/agent/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/agent/__init__.py +++ b/sam3/agent/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/agent/agent_core.py b/sam3/agent/agent_core.py index f0016c7..f893f00 100644 --- a/sam3/agent/agent_core.py +++ b/sam3/agent/agent_core.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import copy import json import os diff --git a/sam3/agent/client_llm.py b/sam3/agent/client_llm.py index 85e513c..c55508e 100644 --- a/sam3/agent/client_llm.py +++ b/sam3/agent/client_llm.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import base64 import os from typing import Any, Optional diff --git a/sam3/agent/client_sam3.py b/sam3/agent/client_sam3.py index d2f64b7..b138e60 100755 --- a/sam3/agent/client_sam3.py +++ b/sam3/agent/client_sam3.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import json import os diff --git a/sam3/agent/helpers/__init__.py b/sam3/agent/helpers/__init__.py index 46d37d2..726cb1e 100755 --- a/sam3/agent/helpers/__init__.py +++ b/sam3/agent/helpers/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/agent/helpers/boxes.py b/sam3/agent/helpers/boxes.py index 6cfef58..40df92e 100755 --- a/sam3/agent/helpers/boxes.py +++ b/sam3/agent/helpers/boxes.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import math from enum import IntEnum, unique from typing import List, Tuple, Union diff --git a/sam3/agent/helpers/color_map.py b/sam3/agent/helpers/color_map.py index 1ea1b29..ef6b615 100644 --- a/sam3/agent/helpers/color_map.py +++ b/sam3/agent/helpers/color_map.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ An awesome colormap for really neat visualizations. Copied from Detectron, and removed gray colors. diff --git a/sam3/agent/helpers/keypoints.py b/sam3/agent/helpers/keypoints.py index 040810f..ae67be1 100755 --- a/sam3/agent/helpers/keypoints.py +++ b/sam3/agent/helpers/keypoints.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Any, List, Tuple, Union import numpy as np diff --git a/sam3/agent/helpers/mask_overlap_removal.py b/sam3/agent/helpers/mask_overlap_removal.py index 386706d..c2c3c68 100644 --- a/sam3/agent/helpers/mask_overlap_removal.py +++ b/sam3/agent/helpers/mask_overlap_removal.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Dict, List import numpy as np diff --git a/sam3/agent/helpers/masks.py b/sam3/agent/helpers/masks.py index dc78140..a55b5e2 100755 --- a/sam3/agent/helpers/masks.py +++ b/sam3/agent/helpers/masks.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import copy import itertools from typing import Any, Iterator, List, Union diff --git a/sam3/agent/helpers/memory.py b/sam3/agent/helpers/memory.py index 5d51f1b..fdb7086 100755 --- a/sam3/agent/helpers/memory.py +++ b/sam3/agent/helpers/memory.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging from contextlib import contextmanager from functools import wraps diff --git a/sam3/agent/helpers/rle.py b/sam3/agent/helpers/rle.py index 277be79..db3faef 100755 --- a/sam3/agent/helpers/rle.py +++ b/sam3/agent/helpers/rle.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Some utilities for RLE encoding that doesn't require downloading the masks to the cpu""" import numpy as np diff --git a/sam3/agent/helpers/roi_align.py b/sam3/agent/helpers/roi_align.py index 640f853..a2f413f 100755 --- a/sam3/agent/helpers/roi_align.py +++ b/sam3/agent/helpers/roi_align.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from torch import nn from torchvision.ops import roi_align diff --git a/sam3/agent/helpers/rotated_boxes.py b/sam3/agent/helpers/rotated_boxes.py index 151a96f..cd39af8 100755 --- a/sam3/agent/helpers/rotated_boxes.py +++ b/sam3/agent/helpers/rotated_boxes.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from __future__ import absolute_import, division, print_function, unicode_literals import math diff --git a/sam3/agent/helpers/som_utils.py b/sam3/agent/helpers/som_utils.py index 0ca96d6..81c6f63 100644 --- a/sam3/agent/helpers/som_utils.py +++ b/sam3/agent/helpers/som_utils.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import colorsys from dataclasses import dataclass from typing import List, Tuple diff --git a/sam3/agent/helpers/visualizer.py b/sam3/agent/helpers/visualizer.py index 3e134d3..c6ce032 100755 --- a/sam3/agent/helpers/visualizer.py +++ b/sam3/agent/helpers/visualizer.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import colorsys import logging import math diff --git a/sam3/agent/helpers/zoom_in.py b/sam3/agent/helpers/zoom_in.py index dc60700..42a8c94 100644 --- a/sam3/agent/helpers/zoom_in.py +++ b/sam3/agent/helpers/zoom_in.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import io import math diff --git a/sam3/agent/inference.py b/sam3/agent/inference.py index 0aac116..85167d1 100644 --- a/sam3/agent/inference.py +++ b/sam3/agent/inference.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import json import os diff --git a/sam3/agent/viz.py b/sam3/agent/viz.py index 286d823..523e246 100644 --- a/sam3/agent/viz.py +++ b/sam3/agent/viz.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import cv2 import numpy as np import pycocotools.mask as mask_utils diff --git a/sam3/eval/__init__.py b/sam3/eval/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/eval/__init__.py +++ b/sam3/eval/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/eval/cgf1_eval.py b/sam3/eval/cgf1_eval.py index 6a0d59f..71fe2ea 100644 --- a/sam3/eval/cgf1_eval.py +++ b/sam3/eval/cgf1_eval.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import contextlib import copy import json diff --git a/sam3/eval/coco_eval.py b/sam3/eval/coco_eval.py index 167f8ef..fbb82a0 100644 --- a/sam3/eval/coco_eval.py +++ b/sam3/eval/coco_eval.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ COCO evaluator that works in distributed mode. diff --git a/sam3/eval/coco_eval_offline.py b/sam3/eval/coco_eval_offline.py index 7b07228..ec8a0c0 100644 --- a/sam3/eval/coco_eval_offline.py +++ b/sam3/eval/coco_eval_offline.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ This evaluator is meant for regular COCO mAP evaluation, for example on the COCO val set. diff --git a/sam3/eval/coco_reindex.py b/sam3/eval/coco_reindex.py index 49cd944..5dc02ab 100644 --- a/sam3/eval/coco_reindex.py +++ b/sam3/eval/coco_reindex.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Self-contained COCO JSON re-indexing function that creates temporary files. """ diff --git a/sam3/eval/coco_writer.py b/sam3/eval/coco_writer.py index 5134e63..54f9307 100644 --- a/sam3/eval/coco_writer.py +++ b/sam3/eval/coco_writer.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ COCO prediction dumper for distributed training. diff --git a/sam3/eval/conversion_util.py b/sam3/eval/conversion_util.py index 68942b6..fb8950b 100644 --- a/sam3/eval/conversion_util.py +++ b/sam3/eval/conversion_util.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import json import os from collections import defaultdict diff --git a/sam3/eval/demo_eval.py b/sam3/eval/demo_eval.py index 70804cc..a6076ad 100644 --- a/sam3/eval/demo_eval.py +++ b/sam3/eval/demo_eval.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ This evaluator is based upon COCO evaluation, but evaluates the model in a "demo" setting. This means that the model's predictions are thresholded and evaluated as "hard" predictions. diff --git a/sam3/eval/hota_eval_toolkit/__init__.py b/sam3/eval/hota_eval_toolkit/__init__.py index 9c0fa90..930069a 100644 --- a/sam3/eval/hota_eval_toolkit/__init__.py +++ b/sam3/eval/hota_eval_toolkit/__init__.py @@ -1 +1,3 @@ # flake8: noqa + +# pyre-unsafe diff --git a/sam3/eval/hota_eval_toolkit/run_ytvis_eval.py b/sam3/eval/hota_eval_toolkit/run_ytvis_eval.py index c39dd05..05654e7 100644 --- a/sam3/eval/hota_eval_toolkit/run_ytvis_eval.py +++ b/sam3/eval/hota_eval_toolkit/run_ytvis_eval.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + """run_youtube_vis.py Run example: run_youtube_vis.py --USE_PARALLEL False --METRICS HOTA --TRACKERS_TO_EVAL STEm_Seg diff --git a/sam3/eval/hota_eval_toolkit/trackeval/__init__.py b/sam3/eval/hota_eval_toolkit/trackeval/__init__.py index 131e2b7..e03bbe1 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/__init__.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/__init__.py @@ -1,4 +1,6 @@ # flake8: noqa +# pyre-unsafe + from . import datasets, metrics, utils from .eval import Evaluator diff --git a/sam3/eval/hota_eval_toolkit/trackeval/_timing.py b/sam3/eval/hota_eval_toolkit/trackeval/_timing.py index ad414dd..23fa33c 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/_timing.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/_timing.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + import inspect from functools import wraps from time import perf_counter diff --git a/sam3/eval/hota_eval_toolkit/trackeval/datasets/__init__.py b/sam3/eval/hota_eval_toolkit/trackeval/datasets/__init__.py index 42f1620..68e617d 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/datasets/__init__.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/datasets/__init__.py @@ -1,4 +1,6 @@ # flake8: noqa +# pyre-unsafe + from .tao_ow import TAO_OW from .youtube_vis import YouTubeVIS diff --git a/sam3/eval/hota_eval_toolkit/trackeval/datasets/_base_dataset.py b/sam3/eval/hota_eval_toolkit/trackeval/datasets/_base_dataset.py index fb68025..49c3c03 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/datasets/_base_dataset.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/datasets/_base_dataset.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + import csv import io import os diff --git a/sam3/eval/hota_eval_toolkit/trackeval/datasets/tao_ow.py b/sam3/eval/hota_eval_toolkit/trackeval/datasets/tao_ow.py index 06bc93b..545a051 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/datasets/tao_ow.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/datasets/tao_ow.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + import itertools import json import os diff --git a/sam3/eval/hota_eval_toolkit/trackeval/datasets/youtube_vis.py b/sam3/eval/hota_eval_toolkit/trackeval/datasets/youtube_vis.py index e611398..7ada4a2 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/datasets/youtube_vis.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/datasets/youtube_vis.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + # note: this file has been modified from its original version in TrackEval in # https://github.com/JonathonLuiten/TrackEval/blob/master/trackeval/datasets/youtube_vis.py # to support the following: diff --git a/sam3/eval/hota_eval_toolkit/trackeval/eval.py b/sam3/eval/hota_eval_toolkit/trackeval/eval.py index d2d7205..ed32ff1 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/eval.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/eval.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + import os import time import traceback diff --git a/sam3/eval/hota_eval_toolkit/trackeval/metrics/__init__.py b/sam3/eval/hota_eval_toolkit/trackeval/metrics/__init__.py index 6c84342..531a085 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/metrics/__init__.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/metrics/__init__.py @@ -1,4 +1,6 @@ # flake8: noqa +# pyre-unsafe + from .count import Count from .hota import HOTA diff --git a/sam3/eval/hota_eval_toolkit/trackeval/metrics/_base_metric.py b/sam3/eval/hota_eval_toolkit/trackeval/metrics/_base_metric.py index afbfaca..bf7bb0a 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/metrics/_base_metric.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/metrics/_base_metric.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + from abc import ABC, abstractmethod import numpy as np diff --git a/sam3/eval/hota_eval_toolkit/trackeval/metrics/count.py b/sam3/eval/hota_eval_toolkit/trackeval/metrics/count.py index a372605..6b84468 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/metrics/count.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/metrics/count.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + from .. import _timing from ._base_metric import _BaseMetric diff --git a/sam3/eval/hota_eval_toolkit/trackeval/metrics/hota.py b/sam3/eval/hota_eval_toolkit/trackeval/metrics/hota.py index 4cd9501..9ae2c96 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/metrics/hota.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/metrics/hota.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + import os import numpy as np diff --git a/sam3/eval/hota_eval_toolkit/trackeval/utils.py b/sam3/eval/hota_eval_toolkit/trackeval/utils.py index 8cdf77e..99dcc4c 100644 --- a/sam3/eval/hota_eval_toolkit/trackeval/utils.py +++ b/sam3/eval/hota_eval_toolkit/trackeval/utils.py @@ -1,5 +1,7 @@ # flake8: noqa +# pyre-unsafe + import argparse import csv import os diff --git a/sam3/eval/postprocessors.py b/sam3/eval/postprocessors.py index 973da11..7bd26a9 100644 --- a/sam3/eval/postprocessors.py +++ b/sam3/eval/postprocessors.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Postprocessors class to transform MDETR output according to the downstream task""" import dataclasses diff --git a/sam3/eval/saco_veval_eval.py b/sam3/eval/saco_veval_eval.py index 4f0ed2b..ff4e923 100644 --- a/sam3/eval/saco_veval_eval.py +++ b/sam3/eval/saco_veval_eval.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import json import os diff --git a/sam3/eval/saco_veval_evaluators.py b/sam3/eval/saco_veval_evaluators.py index 4947472..69a2796 100644 --- a/sam3/eval/saco_veval_evaluators.py +++ b/sam3/eval/saco_veval_evaluators.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import json import os import tempfile diff --git a/sam3/eval/teta_eval_toolkit/__init__.py b/sam3/eval/teta_eval_toolkit/__init__.py index 1609f80..9420ee6 100644 --- a/sam3/eval/teta_eval_toolkit/__init__.py +++ b/sam3/eval/teta_eval_toolkit/__init__.py @@ -1,5 +1,7 @@ # fmt: off # flake8: noqa +# pyre-unsafe + from . import config, datasets, metrics, utils from .eval import Evaluator diff --git a/sam3/eval/teta_eval_toolkit/_timing.py b/sam3/eval/teta_eval_toolkit/_timing.py index 1c6fbfa..72f195d 100644 --- a/sam3/eval/teta_eval_toolkit/_timing.py +++ b/sam3/eval/teta_eval_toolkit/_timing.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + import inspect from functools import wraps from time import perf_counter diff --git a/sam3/eval/teta_eval_toolkit/config.py b/sam3/eval/teta_eval_toolkit/config.py index 6342fa2..003d8f5 100644 --- a/sam3/eval/teta_eval_toolkit/config.py +++ b/sam3/eval/teta_eval_toolkit/config.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + """Config.""" import argparse import os diff --git a/sam3/eval/teta_eval_toolkit/datasets/__init__.py b/sam3/eval/teta_eval_toolkit/datasets/__init__.py index 97087d7..2ef17b5 100644 --- a/sam3/eval/teta_eval_toolkit/datasets/__init__.py +++ b/sam3/eval/teta_eval_toolkit/datasets/__init__.py @@ -1,5 +1,7 @@ # fmt: off # flake8: noqa + +# pyre-unsafe """Datasets.""" from .coco import COCO from .tao import TAO diff --git a/sam3/eval/teta_eval_toolkit/datasets/_base_dataset.py b/sam3/eval/teta_eval_toolkit/datasets/_base_dataset.py index fc2c30c..c5a1551 100644 --- a/sam3/eval/teta_eval_toolkit/datasets/_base_dataset.py +++ b/sam3/eval/teta_eval_toolkit/datasets/_base_dataset.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + import csv import io import os diff --git a/sam3/eval/teta_eval_toolkit/datasets/coco.py b/sam3/eval/teta_eval_toolkit/datasets/coco.py index ca3d823..dcbbf90 100644 --- a/sam3/eval/teta_eval_toolkit/datasets/coco.py +++ b/sam3/eval/teta_eval_toolkit/datasets/coco.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + """COCO Dataset.""" import copy import itertools diff --git a/sam3/eval/teta_eval_toolkit/datasets/tao.py b/sam3/eval/teta_eval_toolkit/datasets/tao.py index 7d2bbcc..63fea8e 100644 --- a/sam3/eval/teta_eval_toolkit/datasets/tao.py +++ b/sam3/eval/teta_eval_toolkit/datasets/tao.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + """TAO Dataset.""" import copy import itertools diff --git a/sam3/eval/teta_eval_toolkit/eval.py b/sam3/eval/teta_eval_toolkit/eval.py index 336f10b..07c0a76 100644 --- a/sam3/eval/teta_eval_toolkit/eval.py +++ b/sam3/eval/teta_eval_toolkit/eval.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + import copy import os import pickle diff --git a/sam3/eval/teta_eval_toolkit/metrics/__init__.py b/sam3/eval/teta_eval_toolkit/metrics/__init__.py index 8352cd4..5c39ac6 100644 --- a/sam3/eval/teta_eval_toolkit/metrics/__init__.py +++ b/sam3/eval/teta_eval_toolkit/metrics/__init__.py @@ -1,4 +1,6 @@ # fmt: off # flake8: noqa +# pyre-unsafe + from .teta import TETA diff --git a/sam3/eval/teta_eval_toolkit/metrics/_base_metric.py b/sam3/eval/teta_eval_toolkit/metrics/_base_metric.py index 521a76d..e1d8f77 100644 --- a/sam3/eval/teta_eval_toolkit/metrics/_base_metric.py +++ b/sam3/eval/teta_eval_toolkit/metrics/_base_metric.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + from abc import ABC, abstractmethod import numpy as np diff --git a/sam3/eval/teta_eval_toolkit/metrics/teta.py b/sam3/eval/teta_eval_toolkit/metrics/teta.py index 329626f..288623e 100644 --- a/sam3/eval/teta_eval_toolkit/metrics/teta.py +++ b/sam3/eval/teta_eval_toolkit/metrics/teta.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + """Track Every Thing Accuracy metric.""" import numpy as np diff --git a/sam3/eval/teta_eval_toolkit/utils.py b/sam3/eval/teta_eval_toolkit/utils.py index aa688e7..933bf7a 100644 --- a/sam3/eval/teta_eval_toolkit/utils.py +++ b/sam3/eval/teta_eval_toolkit/utils.py @@ -1,6 +1,8 @@ # fmt: off # flake8: noqa +# pyre-unsafe + import csv import os from collections import OrderedDict diff --git a/sam3/eval/ytvis_coco_wrapper.py b/sam3/eval/ytvis_coco_wrapper.py index ced5899..5412d69 100644 --- a/sam3/eval/ytvis_coco_wrapper.py +++ b/sam3/eval/ytvis_coco_wrapper.py @@ -1,6 +1,8 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved # (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary. +# pyre-unsafe + import copy import json import logging diff --git a/sam3/eval/ytvis_eval.py b/sam3/eval/ytvis_eval.py index 93f1cd6..2ff2309 100644 --- a/sam3/eval/ytvis_eval.py +++ b/sam3/eval/ytvis_eval.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import copy import gc import logging diff --git a/sam3/logger.py b/sam3/logger.py index db9c0a6..35dcc0d 100644 --- a/sam3/logger.py +++ b/sam3/logger.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import logging import os diff --git a/sam3/model/__init__.py b/sam3/model/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/model/__init__.py +++ b/sam3/model/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/model/act_ckpt_utils.py b/sam3/model/act_ckpt_utils.py index c935cfc..6a6d983 100644 --- a/sam3/model/act_ckpt_utils.py +++ b/sam3/model/act_ckpt_utils.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import inspect from functools import wraps from typing import Callable, TypeVar, Union diff --git a/sam3/model/box_ops.py b/sam3/model/box_ops.py index f88e4ad..59f52e0 100644 --- a/sam3/model/box_ops.py +++ b/sam3/model/box_ops.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe """ Utilities for bounding box manipulation and GIoU. """ diff --git a/sam3/model/data_misc.py b/sam3/model/data_misc.py index 4bbcf55..8f2efa9 100644 --- a/sam3/model/data_misc.py +++ b/sam3/model/data_misc.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe """ Misc functions, including distributed helpers. """ diff --git a/sam3/model/decoder.py b/sam3/model/decoder.py index c8b1657..c074b9c 100644 --- a/sam3/model/decoder.py +++ b/sam3/model/decoder.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe """ Transformer decoder. Inspired from Pytorch's version, adds the pre-norm variant diff --git a/sam3/model/edt.py b/sam3/model/edt.py index 9448c1d..498f28f 100644 --- a/sam3/model/edt.py +++ b/sam3/model/edt.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Triton kernel for euclidean distance transform (EDT)""" import torch diff --git a/sam3/model/encoder.py b/sam3/model/encoder.py index 842bc56..d825df4 100644 --- a/sam3/model/encoder.py +++ b/sam3/model/encoder.py @@ -1,6 +1,8 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved # Based on https://github.com/IDEA-Research/GroundingDINO +# pyre-unsafe + from typing import Any, Dict, List, Optional, Tuple import torch diff --git a/sam3/model/geometry_encoders.py b/sam3/model/geometry_encoders.py index bff2917..acd9a15 100644 --- a/sam3/model/geometry_encoders.py +++ b/sam3/model/geometry_encoders.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Tuple import torch diff --git a/sam3/model/io_utils.py b/sam3/model/io_utils.py index 0a22584..082ba23 100644 --- a/sam3/model/io_utils.py +++ b/sam3/model/io_utils.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import contextlib import os import queue diff --git a/sam3/model/maskformer_segmentation.py b/sam3/model/maskformer_segmentation.py index e1f5ae8..8790d7a 100644 --- a/sam3/model/maskformer_segmentation.py +++ b/sam3/model/maskformer_segmentation.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import math from typing import Dict, List, Optional diff --git a/sam3/model/memory.py b/sam3/model/memory.py index bfde548..397e1c8 100644 --- a/sam3/model/memory.py +++ b/sam3/model/memory.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import math from typing import Tuple diff --git a/sam3/model/model_misc.py b/sam3/model/model_misc.py index 2cb44b3..9fe38a1 100644 --- a/sam3/model/model_misc.py +++ b/sam3/model/model_misc.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Various utility models""" import copy diff --git a/sam3/model/necks.py b/sam3/model/necks.py index 21bf9b8..5a30825 100644 --- a/sam3/model/necks.py +++ b/sam3/model/necks.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Necks are the interface between a vision backbone and the rest of the detection model""" from copy import deepcopy diff --git a/sam3/model/position_encoding.py b/sam3/model/position_encoding.py index eb3f405..a6a1266 100644 --- a/sam3/model/position_encoding.py +++ b/sam3/model/position_encoding.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import math from typing import Optional diff --git a/sam3/model/sam1_task_predictor.py b/sam3/model/sam1_task_predictor.py index f5e49b1..1ca18bd 100644 --- a/sam3/model/sam1_task_predictor.py +++ b/sam3/model/sam1_task_predictor.py @@ -1,6 +1,8 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved # All rights reserved. +# pyre-unsafe + # This source code is licensed under the license found in the # LICENSE file in the root directory of this source tree. diff --git a/sam3/model/sam3_image.py b/sam3/model/sam3_image.py index aafe520..3e4f9de 100644 --- a/sam3/model/sam3_image.py +++ b/sam3/model/sam3_image.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import os from copy import deepcopy from typing import Dict, List, Optional, Tuple diff --git a/sam3/model/sam3_image_processor.py b/sam3/model/sam3_image_processor.py index 4d98fbf..7a55e02 100644 --- a/sam3/model/sam3_image_processor.py +++ b/sam3/model/sam3_image_processor.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe from typing import Dict, List import numpy as np diff --git a/sam3/model/sam3_tracker_base.py b/sam3/model/sam3_tracker_base.py index 90fbd69..a5c557d 100644 --- a/sam3/model/sam3_tracker_base.py +++ b/sam3/model/sam3_tracker_base.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging import torch diff --git a/sam3/model/sam3_tracker_utils.py b/sam3/model/sam3_tracker_utils.py index 7afc70a..e88c093 100644 --- a/sam3/model/sam3_tracker_utils.py +++ b/sam3/model/sam3_tracker_utils.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import numpy as np import torch import torch.nn.functional as F diff --git a/sam3/model/sam3_tracking_predictor.py b/sam3/model/sam3_tracking_predictor.py index b2440ef..0fb3ff7 100644 --- a/sam3/model/sam3_tracking_predictor.py +++ b/sam3/model/sam3_tracking_predictor.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging from collections import OrderedDict diff --git a/sam3/model/sam3_video_base.py b/sam3/model/sam3_video_base.py index e61969f..78e1044 100644 --- a/sam3/model/sam3_video_base.py +++ b/sam3/model/sam3_video_base.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import datetime import logging import math diff --git a/sam3/model/sam3_video_inference.py b/sam3/model/sam3_video_inference.py index 7fb87d0..263b4d2 100644 --- a/sam3/model/sam3_video_inference.py +++ b/sam3/model/sam3_video_inference.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging from collections import defaultdict diff --git a/sam3/model/sam3_video_predictor.py b/sam3/model/sam3_video_predictor.py index c639e1d..de2ec60 100644 --- a/sam3/model/sam3_video_predictor.py +++ b/sam3/model/sam3_video_predictor.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import datetime import gc import multiprocessing as mp diff --git a/sam3/model/text_encoder_ve.py b/sam3/model/text_encoder_ve.py index b1cf145..53ddd5d 100644 --- a/sam3/model/text_encoder_ve.py +++ b/sam3/model/text_encoder_ve.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from collections import OrderedDict from typing import Callable, List, Optional, Tuple, Union diff --git a/sam3/model/tokenizer_ve.py b/sam3/model/tokenizer_ve.py index ef42773..f11fd3d 100644 --- a/sam3/model/tokenizer_ve.py +++ b/sam3/model/tokenizer_ve.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Text Tokenizer. diff --git a/sam3/model/utils/__init__.py b/sam3/model/utils/__init__.py index 47d9858..8fb7069 100644 --- a/sam3/model/utils/__init__.py +++ b/sam3/model/utils/__init__.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved # All rights reserved. +# pyre-unsafe + # This source code is licensed under the license found in the # LICENSE file in the root directory of this source tree. diff --git a/sam3/model/utils/misc.py b/sam3/model/utils/misc.py index 4f07206..486fa8d 100644 --- a/sam3/model/utils/misc.py +++ b/sam3/model/utils/misc.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from collections import defaultdict from dataclasses import fields, is_dataclass from typing import Any, Mapping, Protocol, runtime_checkable diff --git a/sam3/model/utils/sam1_utils.py b/sam3/model/utils/sam1_utils.py index 18f0d04..1ee131e 100644 --- a/sam3/model/utils/sam1_utils.py +++ b/sam3/model/utils/sam1_utils.py @@ -1,6 +1,8 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved # All rights reserved. +# pyre-unsafe + # This source code is licensed under the license found in the # LICENSE file in the root directory of this source tree. diff --git a/sam3/model/utils/sam2_utils.py b/sam3/model/utils/sam2_utils.py index d91ba0f..e824fe1 100644 --- a/sam3/model/utils/sam2_utils.py +++ b/sam3/model/utils/sam2_utils.py @@ -1,6 +1,8 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved # All rights reserved. +# pyre-unsafe + # This source code is licensed under the license found in the # LICENSE file in the root directory of this source tree. diff --git a/sam3/model/vitdet.py b/sam3/model/vitdet.py index aa56664..1b4d41b 100644 --- a/sam3/model/vitdet.py +++ b/sam3/model/vitdet.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ ViTDet backbone adapted from Detectron2. This module implements Vision Transformer (ViT) backbone for object detection. diff --git a/sam3/model/vl_combiner.py b/sam3/model/vl_combiner.py index 43bc7bd..5c400e0 100644 --- a/sam3/model/vl_combiner.py +++ b/sam3/model/vl_combiner.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Provides utility to combine a vision backbone with a language backbone.""" from copy import copy diff --git a/sam3/model_builder.py b/sam3/model_builder.py index 1a3bdec..4df2671 100644 --- a/sam3/model_builder.py +++ b/sam3/model_builder.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import os from typing import Optional diff --git a/sam3/perflib/__init__.py b/sam3/perflib/__init__.py index 5c3823b..9aec2af 100644 --- a/sam3/perflib/__init__.py +++ b/sam3/perflib/__init__.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import os is_enabled = False diff --git a/sam3/perflib/associate_det_trk.py b/sam3/perflib/associate_det_trk.py index 508ae81..7e0d29d 100644 --- a/sam3/perflib/associate_det_trk.py +++ b/sam3/perflib/associate_det_trk.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from collections import defaultdict import torch diff --git a/sam3/perflib/compile.py b/sam3/perflib/compile.py index f427aa7..914c31b 100644 --- a/sam3/perflib/compile.py +++ b/sam3/perflib/compile.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import torch diff --git a/sam3/perflib/connected_components.py b/sam3/perflib/connected_components.py index c96932a..0be67e5 100644 --- a/sam3/perflib/connected_components.py +++ b/sam3/perflib/connected_components.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import logging import torch diff --git a/sam3/perflib/fa3.py b/sam3/perflib/fa3.py index 8f8c9bd..af226bc 100644 --- a/sam3/perflib/fa3.py +++ b/sam3/perflib/fa3.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import torch diff --git a/sam3/perflib/masks_ops.py b/sam3/perflib/masks_ops.py index 48299d5..7946996 100644 --- a/sam3/perflib/masks_ops.py +++ b/sam3/perflib/masks_ops.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import torch diff --git a/sam3/perflib/nms.py b/sam3/perflib/nms.py index b3efc59..1b2f835 100644 --- a/sam3/perflib/nms.py +++ b/sam3/perflib/nms.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging import numpy as np diff --git a/sam3/perflib/tests/tests.py b/sam3/perflib/tests/tests.py index 0fb88ad..f698b6a 100644 --- a/sam3/perflib/tests/tests.py +++ b/sam3/perflib/tests/tests.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import os import numpy as np diff --git a/sam3/perflib/triton/connected_components.py b/sam3/perflib/triton/connected_components.py index 253ca9d..cdb7d44 100644 --- a/sam3/perflib/triton/connected_components.py +++ b/sam3/perflib/triton/connected_components.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import math import torch diff --git a/sam3/perflib/triton/nms.py b/sam3/perflib/triton/nms.py index ed800a1..9a06f33 100644 --- a/sam3/perflib/triton/nms.py +++ b/sam3/perflib/triton/nms.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + # Adapted from https://github.com/stackav-oss/conch/blob/main/conch/kernels/vision/nms.py import torch diff --git a/sam3/sam/__init__.py b/sam3/sam/__init__.py index 8b35da6..e728afd 100644 --- a/sam3/sam/__init__.py +++ b/sam3/sam/__init__.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe from .mask_decoder import MaskDecoder from .prompt_encoder import PromptEncoder from .transformer import TwoWayTransformer diff --git a/sam3/sam/common.py b/sam3/sam/common.py index b6d1587..72b1830 100644 --- a/sam3/sam/common.py +++ b/sam3/sam/common.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Type import torch diff --git a/sam3/sam/mask_decoder.py b/sam3/sam/mask_decoder.py index b4ac397..3e1bbd2 100644 --- a/sam3/sam/mask_decoder.py +++ b/sam3/sam/mask_decoder.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import List, Optional, Tuple, Type import torch diff --git a/sam3/sam/prompt_encoder.py b/sam3/sam/prompt_encoder.py index 145ea9f..b545a2d 100644 --- a/sam3/sam/prompt_encoder.py +++ b/sam3/sam/prompt_encoder.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Any, Optional, Tuple, Type import numpy as np diff --git a/sam3/sam/rope.py b/sam3/sam/rope.py index 2db01b6..6099273 100644 --- a/sam3/sam/rope.py +++ b/sam3/sam/rope.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Adapted from: 1. https://github.com/meta-llama/codellama/blob/main/llama/model.py diff --git a/sam3/sam/transformer.py b/sam3/sam/transformer.py index 3e96c28..5c4bd34 100644 --- a/sam3/sam/transformer.py +++ b/sam3/sam/transformer.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import math from functools import partial from typing import Tuple, Type diff --git a/sam3/train/__init__.py b/sam3/train/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/train/__init__.py +++ b/sam3/train/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/train/data/__init__.py b/sam3/train/data/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/train/data/__init__.py +++ b/sam3/train/data/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/train/data/coco_json_loaders.py b/sam3/train/data/coco_json_loaders.py index 4bdf487..1618e19 100644 --- a/sam3/train/data/coco_json_loaders.py +++ b/sam3/train/data/coco_json_loaders.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import json from collections import defaultdict from typing import Dict, List, Tuple diff --git a/sam3/train/data/collator.py b/sam3/train/data/collator.py index e0f7e2e..b32b7e3 100644 --- a/sam3/train/data/collator.py +++ b/sam3/train/data/collator.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from dataclasses import dataclass, field as field_ptr_behaviour, fields, is_dataclass from typing import Any, get_args, get_origin, List, Union diff --git a/sam3/train/data/sam3_image_dataset.py b/sam3/train/data/sam3_image_dataset.py index 97efb1d..7941e37 100644 --- a/sam3/train/data/sam3_image_dataset.py +++ b/sam3/train/data/sam3_image_dataset.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Dataset class for modulated detection""" import json diff --git a/sam3/train/data/sam3_video_dataset.py b/sam3/train/data/sam3_video_dataset.py index 8ee8a4f..91396f0 100644 --- a/sam3/train/data/sam3_video_dataset.py +++ b/sam3/train/data/sam3_video_dataset.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import copy import io diff --git a/sam3/train/data/torch_dataset.py b/sam3/train/data/torch_dataset.py index 0555617..5ec9f68 100644 --- a/sam3/train/data/torch_dataset.py +++ b/sam3/train/data/torch_dataset.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Callable, Iterable, Optional from torch.utils.data import DataLoader, Dataset, DistributedSampler, IterableDataset diff --git a/sam3/train/loss/__init__.py b/sam3/train/loss/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/train/loss/__init__.py +++ b/sam3/train/loss/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/train/loss/loss_fns.py b/sam3/train/loss/loss_fns.py index 3b61d7f..2bb9039 100644 --- a/sam3/train/loss/loss_fns.py +++ b/sam3/train/loss/loss_fns.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging import warnings diff --git a/sam3/train/loss/mask_sampling.py b/sam3/train/loss/mask_sampling.py index aeba3fe..aa0ad00 100644 --- a/sam3/train/loss/mask_sampling.py +++ b/sam3/train/loss/mask_sampling.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + from typing import Callable import torch diff --git a/sam3/train/loss/sam3_loss.py b/sam3/train/loss/sam3_loss.py index 7ef59dc..300100a 100644 --- a/sam3/train/loss/sam3_loss.py +++ b/sam3/train/loss/sam3_loss.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import torch from sam3.model.model_misc import SAM3Output diff --git a/sam3/train/loss/sigmoid_focal_loss.py b/sam3/train/loss/sigmoid_focal_loss.py index 15e6db4..a56f03a 100644 --- a/sam3/train/loss/sigmoid_focal_loss.py +++ b/sam3/train/loss/sigmoid_focal_loss.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Triton kernel for faster and memory efficient sigmoid focal loss""" import torch diff --git a/sam3/train/masks_ops.py b/sam3/train/masks_ops.py index f9d2fd7..a98e250 100644 --- a/sam3/train/masks_ops.py +++ b/sam3/train/masks_ops.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Utilities for masks manipulation""" import numpy as np diff --git a/sam3/train/matcher.py b/sam3/train/matcher.py index b0b8d62..660e1d0 100644 --- a/sam3/train/matcher.py +++ b/sam3/train/matcher.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Modules to compute the matching cost and solve the corresponding LSAP. """ diff --git a/sam3/train/nms_helper.py b/sam3/train/nms_helper.py index d6378bc..cf19226 100644 --- a/sam3/train/nms_helper.py +++ b/sam3/train/nms_helper.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import warnings from typing import Dict, List diff --git a/sam3/train/optim/__init__.py b/sam3/train/optim/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/train/optim/__init__.py +++ b/sam3/train/optim/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/train/optim/optimizer.py b/sam3/train/optim/optimizer.py index d401b98..7fa9b90 100644 --- a/sam3/train/optim/optimizer.py +++ b/sam3/train/optim/optimizer.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import fnmatch import inspect import itertools diff --git a/sam3/train/optim/schedulers.py b/sam3/train/optim/schedulers.py index 59da840..139c9b4 100644 --- a/sam3/train/optim/schedulers.py +++ b/sam3/train/optim/schedulers.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import math diff --git a/sam3/train/train.py b/sam3/train/train.py index b3e995c..976fde3 100644 --- a/sam3/train/train.py +++ b/sam3/train/train.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging import os import random diff --git a/sam3/train/trainer.py b/sam3/train/trainer.py index ac7c1b5..4d25d92 100644 --- a/sam3/train/trainer.py +++ b/sam3/train/trainer.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import contextlib import fnmatch import gc diff --git a/sam3/train/transforms/__init__.py b/sam3/train/transforms/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/train/transforms/__init__.py +++ b/sam3/train/transforms/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/train/transforms/basic.py b/sam3/train/transforms/basic.py index 3e8cf1f..cdcce33 100644 --- a/sam3/train/transforms/basic.py +++ b/sam3/train/transforms/basic.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Transforms and data augmentation for both image + bbox. """ diff --git a/sam3/train/transforms/basic_for_api.py b/sam3/train/transforms/basic_for_api.py index e0ec2af..c7cc494 100644 --- a/sam3/train/transforms/basic_for_api.py +++ b/sam3/train/transforms/basic_for_api.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Transforms and data augmentation for both image + bbox. """ diff --git a/sam3/train/transforms/filter_query_transforms.py b/sam3/train/transforms/filter_query_transforms.py index 2d6708f..2075838 100644 --- a/sam3/train/transforms/filter_query_transforms.py +++ b/sam3/train/transforms/filter_query_transforms.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging import random diff --git a/sam3/train/transforms/point_sampling.py b/sam3/train/transforms/point_sampling.py index e083fde..f1f8cad 100644 --- a/sam3/train/transforms/point_sampling.py +++ b/sam3/train/transforms/point_sampling.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import cv2 import numpy as np import torch diff --git a/sam3/train/transforms/segmentation.py b/sam3/train/transforms/segmentation.py index 3e97db0..466a109 100644 --- a/sam3/train/transforms/segmentation.py +++ b/sam3/train/transforms/segmentation.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import numpy as np import pycocotools.mask as mask_utils import torch diff --git a/sam3/train/utils/__init__.py b/sam3/train/utils/__init__.py index 46d37d2..726cb1e 100644 --- a/sam3/train/utils/__init__.py +++ b/sam3/train/utils/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/sam3/train/utils/checkpoint_utils.py b/sam3/train/utils/checkpoint_utils.py index 7f2736a..465e006 100644 --- a/sam3/train/utils/checkpoint_utils.py +++ b/sam3/train/utils/checkpoint_utils.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import contextlib import fnmatch diff --git a/sam3/train/utils/distributed.py b/sam3/train/utils/distributed.py index 3c87a91..a91c4bc 100644 --- a/sam3/train/utils/distributed.py +++ b/sam3/train/utils/distributed.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + # Copyright (c) Meta Platforms, Inc. and affiliates. # All rights reserved. diff --git a/sam3/train/utils/logger.py b/sam3/train/utils/logger.py index 127f6c8..0c835a4 100644 --- a/sam3/train/utils/logger.py +++ b/sam3/train/utils/logger.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import atexit import functools import logging diff --git a/sam3/train/utils/train_utils.py b/sam3/train/utils/train_utils.py index 43a8078..10b6929 100644 --- a/sam3/train/utils/train_utils.py +++ b/sam3/train/utils/train_utils.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + import logging import math import os diff --git a/sam3/visualization_utils.py b/sam3/visualization_utils.py index 73398f2..07cd0f7 100644 --- a/sam3/visualization_utils.py +++ b/sam3/visualization_utils.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import json import os import subprocess diff --git a/scripts/eval/gold/eval_sam3.py b/scripts/eval/gold/eval_sam3.py index 59791ee..f08076b 100644 --- a/scripts/eval/gold/eval_sam3.py +++ b/scripts/eval/gold/eval_sam3.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Script to run the evaluator offline given the GTs for SAC-Gold test set and SAM3 model prediction files. It reports CGF1, IL_MCC, PM_F1 metrics for each subset of SAC-Gold test set. diff --git a/scripts/eval/silver/download_fathomnet.py b/scripts/eval/silver/download_fathomnet.py index ab3de2d..d1ea54a 100644 --- a/scripts/eval/silver/download_fathomnet.py +++ b/scripts/eval/silver/download_fathomnet.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import json import os diff --git a/scripts/eval/silver/download_inaturalist.py b/scripts/eval/silver/download_inaturalist.py index e17bfa8..83b70fe 100644 --- a/scripts/eval/silver/download_inaturalist.py +++ b/scripts/eval/silver/download_inaturalist.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import json import shutil diff --git a/scripts/eval/silver/download_preprocess_nga.py b/scripts/eval/silver/download_preprocess_nga.py index 9c287d9..52f616c 100644 --- a/scripts/eval/silver/download_preprocess_nga.py +++ b/scripts/eval/silver/download_preprocess_nga.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import os from functools import partial diff --git a/scripts/eval/silver/download_videos.py b/scripts/eval/silver/download_videos.py index 352f5e9..b8fda76 100644 --- a/scripts/eval/silver/download_videos.py +++ b/scripts/eval/silver/download_videos.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import ast import concurrent.futures import os diff --git a/scripts/eval/silver/extract_frames.py b/scripts/eval/silver/extract_frames.py index 860ab4e..6be7f05 100644 --- a/scripts/eval/silver/extract_frames.py +++ b/scripts/eval/silver/extract_frames.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe """ This file extracts the frames for the frame datasets in SA-CO/Gold and Silver. diff --git a/scripts/eval/silver/preprocess_silver_geode_bdd100k_food_rec.py b/scripts/eval/silver/preprocess_silver_geode_bdd100k_food_rec.py index 81d7fcf..6b0c65f 100644 --- a/scripts/eval/silver/preprocess_silver_geode_bdd100k_food_rec.py +++ b/scripts/eval/silver/preprocess_silver_geode_bdd100k_food_rec.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse from multiprocessing import Pool from pathlib import Path diff --git a/scripts/eval/silver/utils.py b/scripts/eval/silver/utils.py index dafe813..4968ae0 100644 --- a/scripts/eval/silver/utils.py +++ b/scripts/eval/silver/utils.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import json import os import shutil diff --git a/scripts/eval/standalone_cgf1.py b/scripts/eval/standalone_cgf1.py index f6995e3..823100d 100644 --- a/scripts/eval/standalone_cgf1.py +++ b/scripts/eval/standalone_cgf1.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """Simple script to run the CGF1 evaluator given a prediction file and GT file(s). Usage: python standalone_cgf1.py --pred_file --gt_files ... diff --git a/scripts/eval/veval/__init__.py b/scripts/eval/veval/__init__.py index 46d37d2..726cb1e 100644 --- a/scripts/eval/veval/__init__.py +++ b/scripts/eval/veval/__init__.py @@ -1 +1,3 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe diff --git a/scripts/eval/veval/saco_yt1b_annot_update.py b/scripts/eval/veval/saco_yt1b_annot_update.py index 01be012..e88413a 100644 --- a/scripts/eval/veval/saco_yt1b_annot_update.py +++ b/scripts/eval/veval/saco_yt1b_annot_update.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import json import logging diff --git a/scripts/eval/veval/saco_yt1b_downloader.py b/scripts/eval/veval/saco_yt1b_downloader.py index 4a130ab..ee83102 100644 --- a/scripts/eval/veval/saco_yt1b_downloader.py +++ b/scripts/eval/veval/saco_yt1b_downloader.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import logging diff --git a/scripts/eval/veval/saco_yt1b_frame_prep_util.py b/scripts/eval/veval/saco_yt1b_frame_prep_util.py index d5c51df..49c4f79 100644 --- a/scripts/eval/veval/saco_yt1b_frame_prep_util.py +++ b/scripts/eval/veval/saco_yt1b_frame_prep_util.py @@ -1,4 +1,6 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved + +# pyre-unsafe import argparse import logging import os diff --git a/scripts/extract_odinw_results.py b/scripts/extract_odinw_results.py index ebe1579..24618e1 100644 --- a/scripts/extract_odinw_results.py +++ b/scripts/extract_odinw_results.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """This script summarizes odinw results""" """ diff --git a/scripts/extract_roboflow_vl100_results.py b/scripts/extract_roboflow_vl100_results.py index d1f1520..a32e6af 100644 --- a/scripts/extract_roboflow_vl100_results.py +++ b/scripts/extract_roboflow_vl100_results.py @@ -1,5 +1,7 @@ # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved +# pyre-unsafe + """ Script to extract and analyze training results from Roboflow VL100 experiments.