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 |
|