Add more pre-trained style transfer models. (#70)

Added more examples and files/models into `demos/video/style-transfer`.

Author: Iainmon

demos/video/style-transfer/README.md

@@ -7,3 +7,6 @@ python3 neural_style.py export --model saved_models/mosaic.pth --accel
 
 python3 style_transfer_test.py --model-file=models/exports/cpu/mosaic_float16.pt --input-video-file=videos/deer.mp4 --output-video-file=videos/mosaic_deer.mp4 --show-output
 
+
+
+sh export_and_run_model.sh nature_oil_painting_ep3_bt4_sw3e10_cw_1e5

demos/video/style-transfer/cam.py

@@ -0,0 +1,29 @@
+import cv2
+
+def main():
+    # Initialize the webcam (0 is usually the default camera)
+    cap = cv2.VideoCapture(0)
+    if not cap.isOpened():
+        print("Cannot open camera")
+        return
+
+    while True:
+        # Capture frame-by-frame
+        ret, frame = cap.read()
+        if not ret:
+            print("Failed to grab frame")
+            break
+
+        # Display the resulting frame
+        cv2.imshow("Webcam", frame)
+
+        # Press 'q' to exit
+        if cv2.waitKey(1) & 0xFF == ord('q'):
+            break
+
+    # When everything done, release the capture
+    cap.release()
+    cv2.destroyAllWindows()
+
+if __name__ == "__main__":
+    main()

demos/video/style-transfer/export_and_run_model.sh

@@ -8,5 +8,8 @@ MODEL_NAME=$1
 python3 neural_style.py export --model saved_models/${MODEL_NAME}.model --accel \
     || python3 neural_style.py export --model saved_models/${MODEL_NAME}.pth --accel
 
+# nature_oil_painting_ep3_bt4_sw3e10_cw_1e5
 
-python3 style_transfer_test.py --model-file=models/exports/cpu/${MODEL_NAME}_float16.pt --input-video-file=videos/deer.mp4 --show-output
+# python3 style_transfer_test.py --model-file=models/exports/cpu/${MODEL_NAME}_float16.pt --input-video-file=videos/deer.mp4 --show-output
+
+python3 style_transfer_test.py --model-file=models/exports/cpu/${MODEL_NAME}_float16.pt --use-webcam --show-output