Update run.py
Browse files
run.py
CHANGED
@@ -35,7 +35,7 @@ model.to(device).eval()
|
|
35 |
# counter = Count()
|
36 |
|
37 |
def segment(im, interval_s=2):
|
38 |
-
|
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)
|