Make example style transfer working with identity image function.

Author: Iainmon

demos/video/style-transfer/style_transfer.cpp

@@ -208,13 +208,21 @@ int run_webcam_model(torch::jit::Module& module, int cam_index, int max_fps, boo
 
             // output_bgr = to_mat(processed_output);
 
-            cv::Mat output_rgb = to_mat(processed_output);
+            // cv::Mat output_rgb = to_mat(processed_output);
             // output_bgr = output_rgb;
             // cv::cvtColor(output_rgb, output_bgr, cv::COLOR_RGB2BGR);
             // cv::cvtColor(output_rgb, output_bgr, cv::COLOR_RGB2BGR);
 
+            // working?
+            auto out_mps_tensor = mps_tensor.to(torch::kCPU,true);
+            frame_rgb = to_mat(out_mps_tensor);
+            cv::cvtColor(frame_rgb, output_bgr, cv::COLOR_RGB2BGR); 
 
-            cv::cvtColor(frame_rgb, output_bgr, cv::COLOR_RGB2BGR);
+
+
+            // // works
+            // frame_rgb = to_mat(input_tensor);
+            // cv::cvtColor(frame_rgb, output_bgr, cv::COLOR_RGB2BGR);