vinno97 / realtime-pytorch-cyclegan-and-pix2pix Goto Github PK

Background:
Currently the command line option for videosource accepts type int which makes things difficult for using video files. But the VideoCapture() class from opencv can accept integers and the file path for a video source.

TODO:
The simple solution is to change the command line option and then do the type checking for either a valid file or an integer.

Hi! I was trying to use test_video.py on a file, but i'm using Google Colab so I had to do some modifications to your code...

It read frames but the data it returned didn't look right, for exemple:

`{'Real_A': tensor([[[[-1., -1., -1.],
[-1., -1., -1.],
[-1., -1., -1.],
...,
[-1., -1., -1.],
[-1., -1., -1.],
[-1., -1., -1.]],

     [[-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.],
      ...,
      [-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.]],

     [[-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.],
      ...,
      [-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.]],

     ...,

     [[-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.],
      ...,
      [-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.]],

     [[-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.],
      ...,
      [-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.]],

     [[-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.],
      ...,
      [-1., -1., -1.],
      [-1., -1., -1.],
      [-1., -1., -1.]]]])}

Max value of tensor: tensor(1.)
Min value of tensor: tensor(-1.)`

It isn't returning any more data than this, which means, no B data, or Fake_B data

What could be causing this and how can I fix this?

Thanks, amazing project

Background:
This is a fork from the original pytorch-CycleGAN-and-pix2pix. The original has a bunch of new updates and improved code so it would be nice to update this fork so that the realtime video can use these new features. Luckily I have already patched the realtime feature into the latest release of pytorch-CycleGAN-and-pix2pix, so it should be pretty easy to bring the latest changes of the original into this fork.

TODO:

• Sync this fork with the original
• Change code so that realtime video feature still works