├── .gitignore ├── LICENSE ├── README.md ├── changepth.py ├── checkpoints └── empty.py ├── images ├── 1.png ├── 2.png ├── 3.png ├── paper │ ├── compare_tab1.PNG │ ├── compare_tab2.PNG │ ├── everything.PNG │ ├── everything2.PNG │ ├── frame.png │ ├── frame_00.png │ ├── intro.png │ ├── process.png │ ├── prompt.PNG │ ├── split.PNG │ └── title.PNG └── truck.jpg ├── inference.py ├── load_sam_json.py ├── prune_distill_step1.py ├── prune_distill_step2.py ├── prune_funcs.py ├── segment_anything ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── automatic_mask_generator.cpython-38.pyc │ ├── build_sam.cpython-38.pyc │ └── predictor.cpython-38.pyc ├── automatic_mask_generator.py ├── build_sam.py ├── modeling │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── common.cpython-38.pyc │ │ ├── image_encoder.cpython-38.pyc │ │ ├── mask_decoder.cpython-38.pyc │ │ ├── prompt_encoder.cpython-38.pyc │ │ ├── sam.cpython-38.pyc │ │ └── transformer.cpython-38.pyc │ ├── common.py │ ├── image_encoder.py │ ├── mask_decoder.py │ ├── prompt_encoder.py │ ├── sam.py │ └── transformer.py ├── predictor.py └── utils │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── amg.cpython-38.pyc │ └── transforms.cpython-38.pyc │ ├── amg.py │ ├── onnx.py │ └── transforms.py ├── segment_anything_kd ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── automatic_mask_generator.cpython-38.pyc │ ├── build_sam.cpython-38.pyc │ └── predictor.cpython-38.pyc ├── automatic_mask_generator.py ├── build_sam.py ├── modeling │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── common.cpython-38.pyc │ │ ├── image_encoder.cpython-38.pyc │ │ ├── mask_decoder.cpython-38.pyc │ │ ├── prompt_encoder.cpython-38.pyc │ │ ├── sam.cpython-38.pyc │ │ └── transformer.cpython-38.pyc │ ├── common.py │ ├── image_encoder.py │ ├── mask_decoder.py │ ├── prompt_encoder.py │ ├── sam.py │ └── transformer.py ├── predictor.py └── utils │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── amg.cpython-38.pyc │ └── transforms.cpython-38.pyc │ ├── amg.py │ ├── onnx.py │ └── transforms.py ├── setup.py └── torch_pruning ├── __init__.py ├── __pycache__ ├── __init__.cpython-38.pyc ├── _helpers.cpython-38.pyc ├── dependency.cpython-38.pyc ├── ops.cpython-38.pyc └── serialization.cpython-38.pyc ├── _helpers.py ├── dependency.py ├── ops.py ├── pruner ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── function.cpython-38.pyc │ └── importance.cpython-38.pyc ├── algorithms │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── batchnorm_scale_pruner.cpython-38.pyc │ │ ├── group_norm_pruner.cpython-38.pyc │ │ ├── growing_reg_pruner.cpython-38.pyc │ │ ├── magnitude_based_pruner.cpython-38.pyc │ │ ├── metapruner.cpython-38.pyc │ │ └── scheduler.cpython-38.pyc │ ├── batchnorm_scale_pruner.py │ ├── group_norm_pruner.py │ ├── growing_reg_pruner.py │ ├── magnitude_based_pruner.py │ ├── metapruner.py │ └── scheduler.py ├── function.py └── importance.py ├── serialization.py └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-38.pyc ├── op_counter.cpython-38.pyc └── utils.cpython-38.pyc ├── op_counter.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | 2 | *.pth 3 | record/ 4 | __pycache__/ -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/README.md -------------------------------------------------------------------------------- /changepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/changepth.py -------------------------------------------------------------------------------- /checkpoints/empty.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /images/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/1.png -------------------------------------------------------------------------------- /images/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/2.png -------------------------------------------------------------------------------- /images/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/3.png -------------------------------------------------------------------------------- /images/paper/compare_tab1.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/compare_tab1.PNG -------------------------------------------------------------------------------- /images/paper/compare_tab2.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/compare_tab2.PNG -------------------------------------------------------------------------------- /images/paper/everything.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/everything.PNG -------------------------------------------------------------------------------- /images/paper/everything2.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/everything2.PNG -------------------------------------------------------------------------------- /images/paper/frame.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/frame.png -------------------------------------------------------------------------------- /images/paper/frame_00.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/frame_00.png -------------------------------------------------------------------------------- /images/paper/intro.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/intro.png -------------------------------------------------------------------------------- /images/paper/process.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/process.png -------------------------------------------------------------------------------- /images/paper/prompt.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/prompt.PNG -------------------------------------------------------------------------------- /images/paper/split.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/split.PNG -------------------------------------------------------------------------------- /images/paper/title.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/paper/title.PNG -------------------------------------------------------------------------------- /images/truck.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/images/truck.jpg -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/inference.py -------------------------------------------------------------------------------- /load_sam_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/load_sam_json.py -------------------------------------------------------------------------------- /prune_distill_step1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/prune_distill_step1.py -------------------------------------------------------------------------------- /prune_distill_step2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/prune_distill_step2.py -------------------------------------------------------------------------------- /prune_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/prune_funcs.py -------------------------------------------------------------------------------- /segment_anything/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/__init__.py -------------------------------------------------------------------------------- /segment_anything/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/__pycache__/automatic_mask_generator.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/__pycache__/automatic_mask_generator.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/__pycache__/build_sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/__pycache__/build_sam.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/__pycache__/predictor.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/__pycache__/predictor.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/automatic_mask_generator.py -------------------------------------------------------------------------------- /segment_anything/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/build_sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__init__.py -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/common.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/common.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/image_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/image_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/mask_decoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/mask_decoder.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/prompt_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/prompt_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/sam.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/__pycache__/transformer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/__pycache__/transformer.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/common.py -------------------------------------------------------------------------------- /segment_anything/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/image_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/mask_decoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/modeling/transformer.py -------------------------------------------------------------------------------- /segment_anything/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/predictor.py -------------------------------------------------------------------------------- /segment_anything/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/__init__.py -------------------------------------------------------------------------------- /segment_anything/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/utils/__pycache__/amg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/__pycache__/amg.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/utils/__pycache__/transforms.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/__pycache__/transforms.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/amg.py -------------------------------------------------------------------------------- /segment_anything/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/onnx.py -------------------------------------------------------------------------------- /segment_anything/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything/utils/transforms.py -------------------------------------------------------------------------------- /segment_anything_kd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/__init__.py -------------------------------------------------------------------------------- /segment_anything_kd/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/__pycache__/automatic_mask_generator.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/__pycache__/automatic_mask_generator.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/__pycache__/build_sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/__pycache__/build_sam.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/__pycache__/predictor.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/__pycache__/predictor.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/automatic_mask_generator.py -------------------------------------------------------------------------------- /segment_anything_kd/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/build_sam.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__init__.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/common.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/common.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/image_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/image_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/mask_decoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/mask_decoder.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/prompt_encoder.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/prompt_encoder.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/sam.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/sam.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/__pycache__/transformer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/__pycache__/transformer.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/common.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/image_encoder.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/mask_decoder.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/sam.py -------------------------------------------------------------------------------- /segment_anything_kd/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/modeling/transformer.py -------------------------------------------------------------------------------- /segment_anything_kd/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/predictor.py -------------------------------------------------------------------------------- /segment_anything_kd/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/__init__.py -------------------------------------------------------------------------------- /segment_anything_kd/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/utils/__pycache__/amg.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/__pycache__/amg.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/utils/__pycache__/transforms.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/__pycache__/transforms.cpython-38.pyc -------------------------------------------------------------------------------- /segment_anything_kd/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/amg.py -------------------------------------------------------------------------------- /segment_anything_kd/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/onnx.py -------------------------------------------------------------------------------- /segment_anything_kd/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/segment_anything_kd/utils/transforms.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/setup.py -------------------------------------------------------------------------------- /torch_pruning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/__init__.py -------------------------------------------------------------------------------- /torch_pruning/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/__pycache__/_helpers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/__pycache__/_helpers.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/__pycache__/dependency.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/__pycache__/dependency.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/__pycache__/ops.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/__pycache__/ops.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/__pycache__/serialization.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/__pycache__/serialization.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/_helpers.py -------------------------------------------------------------------------------- /torch_pruning/dependency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/dependency.py -------------------------------------------------------------------------------- /torch_pruning/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/ops.py -------------------------------------------------------------------------------- /torch_pruning/pruner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/__init__.py -------------------------------------------------------------------------------- /torch_pruning/pruner/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/__pycache__/function.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/__pycache__/function.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/__pycache__/importance.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/__pycache__/importance.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__init__.py -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/batchnorm_scale_pruner.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/batchnorm_scale_pruner.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/group_norm_pruner.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/group_norm_pruner.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/growing_reg_pruner.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/growing_reg_pruner.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/magnitude_based_pruner.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/magnitude_based_pruner.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/metapruner.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/metapruner.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/__pycache__/scheduler.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/__pycache__/scheduler.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/batchnorm_scale_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/batchnorm_scale_pruner.py -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/group_norm_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/group_norm_pruner.py -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/growing_reg_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/growing_reg_pruner.py -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/magnitude_based_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/magnitude_based_pruner.py -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/metapruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/metapruner.py -------------------------------------------------------------------------------- /torch_pruning/pruner/algorithms/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/algorithms/scheduler.py -------------------------------------------------------------------------------- /torch_pruning/pruner/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/function.py -------------------------------------------------------------------------------- /torch_pruning/pruner/importance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/pruner/importance.py -------------------------------------------------------------------------------- /torch_pruning/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/serialization.py -------------------------------------------------------------------------------- /torch_pruning/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/utils/__init__.py -------------------------------------------------------------------------------- /torch_pruning/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/utils/__pycache__/op_counter.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/utils/__pycache__/op_counter.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/utils/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/utils/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /torch_pruning/utils/op_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/utils/op_counter.py -------------------------------------------------------------------------------- /torch_pruning/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/czg1225/SlimSAM/HEAD/torch_pruning/utils/utils.py --------------------------------------------------------------------------------