Selaa lähdekoodia

fix a bug in demo

yjh0410 1 vuosi sitten
vanhempi
sitoutus
9a2607ddf0
1 muutettua tiedostoa jossa 3 lisäystä ja 3 poistoa
  1. 3 3
      demo.py

+ 3 - 3
demo.py

@@ -110,7 +110,7 @@ def detect(args,
 
                 # prepare
                 x, _, ratio = transform(frame)
-                x = x.unsqueeze(0).to(device) / 255.
+                x = x.unsqueeze(0).to(device)
                 
                 # inference
                 t0 = time.time()
@@ -179,7 +179,7 @@ def detect(args,
 
                 # prepare
                 x, _, ratio = transform(frame)
-                x = x.unsqueeze(0).to(device) / 255.
+                x = x.unsqueeze(0).to(device)
 
                 # inference
                 t0 = time.time()
@@ -236,7 +236,7 @@ def detect(args,
 
             # prepare
             x, _, ratio = transform(image)
-            x = x.unsqueeze(0).to(device) / 255.
+            x = x.unsqueeze(0).to(device)
 
             # inference
             t0 = time.time()