Update model/inference_cpu.py
Browse files- model/inference_cpu.py +1 -1
model/inference_cpu.py
CHANGED
|
@@ -120,7 +120,7 @@ def zoom_in_zoom_out(args, segvol_model, image, image_resize, text_prompt, point
|
|
| 120 |
def build_model():
|
| 121 |
# build model
|
| 122 |
clip_ckpt = 'model/config/clip'
|
| 123 |
-
resume = '
|
| 124 |
sam_model = sam_model_registry['vit']()
|
| 125 |
segvol_model = SegVol(
|
| 126 |
image_encoder=sam_model.image_encoder,
|
|
|
|
| 120 |
def build_model():
|
| 121 |
# build model
|
| 122 |
clip_ckpt = 'model/config/clip'
|
| 123 |
+
resume = 'SegVol_v1.pth'
|
| 124 |
sam_model = sam_model_registry['vit']()
|
| 125 |
segvol_model = SegVol(
|
| 126 |
image_encoder=sam_model.image_encoder,
|