updated to new code

Author: culurciello

visor/visor.py

@@ -3,7 +3,7 @@
 import sys
 import os
 import time
-import cv2 # install cv3, python3:  http://seeb0h.github.io/howto/howto-install-homebrew-python-opencv-osx-el-capitan/
+import cv2 # install cv3, python3: brew install opencv3 --with-contrib --with-python3 --without-python
 # add to profile: export PYTHONPATH=$PYTHONPATH:/usr/local/Cellar/opencv3/3.2.0/lib/python3.6/site-packages/
 import numpy as np
 import argparse
@@ -82,10 +82,10 @@ def cat_file():
     pframe = cv2.resize(frame, dsize=(args.size, args.size))
 
     # prepare and normalize frame for processing:
-    pframe = np.swapaxes(pframe, 0, 2)
-    pframe = np.expand_dims(pframe, axis=0)
     pframe = transformsImage(pframe)
     pframe = torch.autograd.Variable(pframe) # turn Tensor to variable required for pytorch processing
+    pframe = pframe.unsqueeze(0)
+
 
     # process via CNN model:
     output = model(pframe)