Skip to content

Commit 9a2607d

Browse files
committed
fix a bug in demo
1 parent d106b10 commit 9a2607d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

demo.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def detect(args,
110110

111111
# prepare
112112
x, _, ratio = transform(frame)
113-
x = x.unsqueeze(0).to(device) / 255.
113+
x = x.unsqueeze(0).to(device)
114114

115115
# inference
116116
t0 = time.time()
@@ -179,7 +179,7 @@ def detect(args,
179179

180180
# prepare
181181
x, _, ratio = transform(frame)
182-
x = x.unsqueeze(0).to(device) / 255.
182+
x = x.unsqueeze(0).to(device)
183183

184184
# inference
185185
t0 = time.time()
@@ -236,7 +236,7 @@ def detect(args,
236236

237237
# prepare
238238
x, _, ratio = transform(image)
239-
x = x.unsqueeze(0).to(device) / 255.
239+
x = x.unsqueeze(0).to(device)
240240

241241
# inference
242242
t0 = time.time()

0 commit comments

Comments
 (0)