path	# lines of code
training/dataset/vos_dataset.py	120
training/dataset/__init__.py	1
training/dataset/sam2_datasets.py	116
training/dataset/vos_segment_loader.py	194
training/dataset/vos_sampler.py	67
training/dataset/utils.py	52
training/dataset/transforms.py	426
training/dataset/vos_raw_dataset.py	229
training/model/sam2.py	416
training/model/__init__.py	1
training/utils/checkpoint_utils.py	208
training/utils/__init__.py	1
training/utils/data_utils.py	102
training/utils/train_utils.py	202
training/utils/distributed.py	333
training/utils/logger.py	136
training/scripts/sav_frame_extraction_submitit.py	128
training/__init__.py	1
training/trainer.py	826
training/train.py	212
training/loss_fns.py	194
training/optimizer.py	312
pyproject.toml	6
setup.py	138
sav_dataset/sav_evaluator.py	44
sav_dataset/utils/sav_utils.py	113
sav_dataset/utils/sav_benchmark.py	318
coreml/export.py	422
sam2/modeling/sam2_base.py	640
sam2/modeling/sam2_utils.py	193
sam2/modeling/__init__.py	1
sam2/modeling/memory_encoder.py	128
sam2/modeling/position_encoding.py	160
sam2/modeling/backbones/__init__.py	1
sam2/modeling/backbones/hieradet.py	230
sam2/modeling/backbones/image_encoder.py	94
sam2/modeling/backbones/utils.py	44
sam2/modeling/sam/__init__.py	1
sam2/modeling/sam/mask_decoder.py	202
sam2/modeling/sam/transformer.py	236
sam2/modeling/sam/prompt_encoder.py	133
sam2/modeling/memory_attention.py	130
sam2/configs/sam2.1_training/sam2.1_hiera_b+_MOSE_finetune.yaml	293
sam2/configs/sam2.1/sam2.1_hiera_s.yaml	104
sam2/configs/sam2.1/sam2.1_hiera_t.yaml	104
sam2/configs/sam2.1/sam2.1_hiera_l.yaml	105
sam2/configs/sam2.1/sam2.1_hiera_b+.yaml	101
sam2/configs/sam2/sam2_hiera_l.yaml	102
sam2/configs/sam2/sam2_hiera_b+.yaml	98
sam2/configs/sam2/sam2_hiera_t.yaml	101
sam2/configs/sam2/sam2_hiera_s.yaml	101
sam2/utils/__init__.py	1
sam2/utils/amg.py	233
sam2/utils/misc.py	243
sam2/utils/transforms.py	78
sam2/__init__.py	4
sam2/sam2_hiera_l.yaml	102
sam2/sam2_image_predictor.py	362
sam2/csrc/connected_components.cu	241
sam2/build_sam.py	117
sam2/sam2_hiera_b+.yaml	98
sam2/automatic_mask_generator.py	292
sam2/sam2_hiera_t.yaml	101
sam2/sam2_hiera_s.yaml	101
sam2/sam2_video_predictor.py	855