64bbca62322389c2ac3a4677b152de593ab4f9f5,test_video.py,,,#,15

Before Change

                image = image.cuda()

            out = model(image)
            out_img = out.cpu().data[0].numpy()
            out_img = cv2.cvtColor(np.asarray(out_img), cv2.COLOR_RGB2BGR)

            if IS_REAL_TIME:

After Change

        // read frame
        success, frame = videoCapture.read()
        while success:
            image = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
            image = Variable(ToTensor()(image)).unsqueeze(dim=0)
            if torch.cuda.is_available():
                image = image.cuda()

Instances