zjr commited on
Commit
863eac9
1 Parent(s): c4b5ce2

Fix save dir not found

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ ckpt_url_map = {
59
  'vit_l': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth',
60
  'vit_h': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'
61
  }
62
-
63
  args = parse_augment()
64
 
65
  checkpoint_url = ckpt_url_map[seg_model_map[args.segmenter]]
 
59
  'vit_l': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth',
60
  'vit_h': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'
61
  }
62
+ os.makedirs('result', exist_ok=True)
63
  args = parse_augment()
64
 
65
  checkpoint_url = ckpt_url_map[seg_model_map[args.segmenter]]