Unity reflections fix (#54)

* Added shader for reflections from newest commit to fix problem with freeze * CI: Add build --------- Co-authored-by: Oldrich Krupauer <oldrich.krupauer@pm.me> Co-authored-by: ItsOKayCZ <ItsOKayCZ@users.noreply.github.com>
ItsOKayCZ · Oct 25, 2024 · b1d3c6f · b1d3c6f
1 parent 3db2aa4
commit b1d3c6f
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 3 deletions.
diff --git a/python_trainer/build/StandaloneLinux64/selfDriving_Data/Managed/Assembly-CSharp.dll b/python_trainer/build/StandaloneLinux64/selfDriving_Data/Managed/Assembly-CSharp.dll
diff --git a/python_trainer/build/StandaloneLinux64/selfDriving_Data/globalgamemanagers b/python_trainer/build/StandaloneLinux64/selfDriving_Data/globalgamemanagers
diff --git a/python_trainer/build/StandaloneLinux64/selfDriving_Data/sharedassets0.assets b/python_trainer/build/StandaloneLinux64/selfDriving_Data/sharedassets0.assets
diff --git a/python_trainer/network.py b/python_trainer/network.py
@@ -25,7 +25,7 @@ def __init__(self, visual_input_shape, nonvis_input_shape, device):
         self.device = device
 
         conv_1_args = {"kernels": 16, "kernel_size": 5, "stride": 1}
-        conv_2_args = {"kernels": 32, "kernel_size": 3, "stride": 1}
+        conv_2_args = {"kernels": 8, "kernel_size": 3, "stride": 1}
         nonvis_dense_size = 8
         encoding_size = 126
 

diff --git a/python_trainer/variables.py b/python_trainer/variables.py
@@ -9,7 +9,7 @@
 REDUCE_TEMPERATURE = 1 / 25
 DISCOUNT = 0.95  # devalues future reward
 LEARNING_RATE = 0.0005
-NUM_TRAINING_EXAMPLES = 10000
+NUM_TRAINING_EXAMPLES = 5000
 MAX_TRAINED_EPOCHS = 500
 
 # Reward

diff --git a/unity_env/Assets/Materials/Road/Reflection.shader b/unity_env/Assets/Materials/Road/Reflection.shader
@@ -187,4 +187,4 @@ Shader "Custom/Reflection"
         ENDCG
     }
     FallBack "Diffuse"
-}
+}
diff --git a/unity_env/Assets/Scripts/TrainingReplicator.cs b/unity_env/Assets/Scripts/TrainingReplicator.cs
@@ -82,6 +82,20 @@ void Start()
             ChangeCameraBackgroundColor();
         }
 
+        void SetMaterial()
+        {
+            foreach (var mat in roadMaterials)
+            {
+                mat.SetFloat(
+                    "_ReflectionStrength",
+                    DataChannel.getParameter("reflectionStrength", 1f)
+                );
+                mat.SetFloat("_NoiseScaleX", DataChannel.getParameter("noiseScaleX", 0.4f));
+                mat.SetFloat("_NoiseScaleY", DataChannel.getParameter("noiseScaleY", 0.05f));
+                mat.SetFloat("_Speed", DataChannel.getParameter("noiseSpeed", 0.3f));
+            }
+        }
+
         void Update()
         {
             if (randomBackgroundColor)
-Original file line number
+Diff line change
@@ Expand Up / @@ -187,4 +187,4 @@ Shader "Custom/Reflection" @@
             ENDCG
         }
         FallBack "Diffuse"
-    }
+    }