Hugging Face's logo

Spaces:
MarcHabib
/
segmentation_yolo_sam
Running

App Files Community
MarcHabib commited on
Commit
09e3ceb
·
1 Parent(s): 8166ca3

Updated files

Browse files
Files changed (6) hide show
  1. app.py +2 -2
  2. models/best.pt +3 -0
  3. sam2/segment-anything-2 +1 -0
  4. utils/__pycache__/__init__.cpython-311.pyc +0 -0
  5. utils/__pycache__/helpers.cpython-311.pyc +0 -0
  6. utils/__pycache__/predictor.cpython-311.pyc +0 -0
app.py CHANGED
@@ -86,8 +86,8 @@ load_training_status()
86
  os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
87
 
88
  # Initialize SAM Predictor
89
- MODEL_CFG = r"\sam2\configs\sam2.1\sam2.1_hiera_l.yaml"
90
- CHECKPOINT = r"\checkpoints\sam2.1_hiera_large.pt"
91
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
92
  predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)
93
 
 
86
  os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
87
 
88
  # Initialize SAM Predictor
89
+ MODEL_CFG = r"configs\sam2.1\sam2.1_hiera_l.yaml"
90
+ CHECKPOINT = r"checkpoints\sam2.1_hiera_large.pt"
91
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
92
  predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)
93
 
models/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c1a69f8b7ebf89dff6c8512342a91782165b0536aafdc8ffa40ded5bd647843
3
+ size 6012957
sam2/segment-anything-2 ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit c2ec8e14a185632b0a5d8b161928ceb50197eddc
utils/__pycache__/__init__.cpython-311.pyc CHANGED
Binary files a/utils/__pycache__/__init__.cpython-311.pyc and b/utils/__pycache__/__init__.cpython-311.pyc differ
 
utils/__pycache__/helpers.cpython-311.pyc CHANGED
Binary files a/utils/__pycache__/helpers.cpython-311.pyc and b/utils/__pycache__/helpers.cpython-311.pyc differ
 
utils/__pycache__/predictor.cpython-311.pyc CHANGED
Binary files a/utils/__pycache__/predictor.cpython-311.pyc and b/utils/__pycache__/predictor.cpython-311.pyc differ