noamholz commited on
Commit
4e9033c
·
verified ·
1 Parent(s): cc35efa

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -35,7 +35,7 @@ model.to(device).eval()
35
  # counter = Count()
36
 
37
  def segment(im, interval_s=2):
38
- print(im)
39
  # if (counter.imout.sum() == 0) or ((cnt % 100) == 0):
40
  pixel_values = image_processor(im, return_tensors="pt").pixel_values.to(device)
41
  outputs = model(pixel_values=pixel_values)
 
35
  # counter = Count()
36
 
37
  def segment(im, interval_s=2):
38
+ im = cv2.resize(im, (im.shape[1] // 2, im.shape[0] // 2))
39
  # if (counter.imout.sum() == 0) or ((cnt % 100) == 0):
40
  pixel_values = image_processor(im, return_tensors="pt").pixel_values.to(device)
41
  outputs = model(pixel_values=pixel_values)