flaviagiammarino
commited on
Commit
•
08261c6
1
Parent(s):
5c5e23e
Upload pt_model.py
Browse files- scripts/pt_model.py +2 -1
scripts/pt_model.py
CHANGED
@@ -16,12 +16,13 @@ hf_model = SamModel(config=SamConfig())
|
|
16 |
hf_model.load_state_dict(hf_state_dict)
|
17 |
hf_model.save_pretrained('./')
|
18 |
|
19 |
-
# update the processor
|
20 |
hf_processor = SamProcessor(
|
21 |
image_processor=SamImageProcessor(
|
22 |
do_normalize=False,
|
23 |
image_mean=[0, 0, 0],
|
24 |
image_std=[1, 1, 1],
|
|
|
25 |
)
|
26 |
)
|
27 |
|
|
|
16 |
hf_model.load_state_dict(hf_state_dict)
|
17 |
hf_model.save_pretrained('./')
|
18 |
|
19 |
+
# update the processor
|
20 |
hf_processor = SamProcessor(
|
21 |
image_processor=SamImageProcessor(
|
22 |
do_normalize=False,
|
23 |
image_mean=[0, 0, 0],
|
24 |
image_std=[1, 1, 1],
|
25 |
+
resample=3,
|
26 |
)
|
27 |
)
|
28 |
|