Skip to content

Commit d29fa74

Browse files
authored
Update demo_video.py
change rgb order
1 parent 03306c5 commit d29fa74

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/demo_video.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ def get_func(inpth, in_q):
5353
while cap.isOpened():
5454
ret, frame = cap.read()
5555
if not ret: break
56+
frame = frame[:, :, ::-1]
5657
frame = to_tensor(dict(im=frame, lb=None))['im'].unsqueeze(0)
57-
frame = frame.flip(dims=(1,)) # rgb
5858
in_q.put(frame)
5959

6060
in_q.put('quit')

0 commit comments

Comments
 (0)