Style transfer C++ demo working completely with mosaic. Need to optimize performance.

Author: Iainmon

demos/video/include/cvtool.hpp

@@ -223,12 +223,14 @@ cv::Mat to_mat(at::Tensor &tensor, cv::ColorConversionCodes color_conversion) {
     int height = tensor.size(2);
     int width = tensor.size(3);
     auto t = tensor
-                .mul(255)
+                .to(torch::kFloat32)
+                .mul(255.0)
+                .clamp(0.0, 255.0)
+                .to(torch::kUInt8)
                 .squeeze()
                 .detach()
                 .permute({1, 2, 0})
                 .contiguous()
-                .to(torch::kUInt8)
                 // .clamp(0, 255)
                 .clone()
                 // .to(cvtool::get_default_device(), /*non_blocking=*/true, /*copy=*/true)

demos/video/style-transfer/style_transfer.cpp

@@ -225,9 +225,9 @@ int run_webcam_model(torch::jit::Module& module, int cam_index, int max_fps, boo
 
 
             // // // works
-            auto processed_input = run_model(module,prepped_input.to(torch::kFloat16)) / 255.0;
-            auto out_processed_input = processed_input; // processed_input.to(torch::kCPU,true);
-            output_bgr = to_mat(out_processed_input, cv::COLOR_RGB2BGR);
+            auto input = input_tensor.to(device,true).to(torch::kFloat16) / 255.0;
+            auto model_output = run_model(module,input) / 255.0;
+            output_bgr = to_mat(model_output, cv::COLOR_RGB2BGR);
 
             // // works
             // auto processed_input = prepped_input;