flaviagiammarino commited on
Commit
08261c6
1 Parent(s): 5c5e23e

Upload pt_model.py

Browse files
Files changed (1) hide show
  1. 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, inputs are min-max scaled instead of normalized
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