Add missing Pyre mode headers] [batch:11/N] [shard:17/N]

Differential Revision: D90237984

fbshipit-source-id: 526fd760f303bf31be4f743bdcd77760496de0de

sam3/train/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

sam3/train/data/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

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

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
 

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

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

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

sam3/train/loss/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

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
 

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

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

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

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

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.
 """

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
 

sam3/train/optim/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

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

sam3/train/optim/schedulers.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import math
 
 

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

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

sam3/train/transforms/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

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.
 """

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.
 """

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
 

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

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

sam3/train/utils/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

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

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.
 

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

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