distrortion fix

allenai · Sep 19, 2024 · 630924d · 630924d
1 parent 9fdd254
commit 630924d
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/unity/Assets/Scripts/AgentManager.cs b/unity/Assets/Scripts/AgentManager.cs
@@ -1374,6 +1374,20 @@ bool shouldRenderImageSynthesis
                         "_flow",
                         "image_thirdParty_flow"
                     ); // XXX fix this in a bit
+                    addImageSynthesisImage(
+                        renderPayload,
+                        imageSynthesis,
+                        this.renderSemanticSegmentation,
+                        "_distortion",
+                        "image_thirdParty_distortion"
+                    );
+                    addImageSynthesisImage(
+                        renderPayload,
+                        imageSynthesis,
+                        this.renderDistortionImage,
+                        "_distortion",
+                        "image_thirdParty_distortion"
+                    );
                 }
             }
         }
@@ -1420,6 +1434,13 @@ bool shouldRenderImageSynthesis
                         "_flow",
                         "image_flow"
                     );
+                    addImageSynthesisImage(
+                        renderPayload,
+                        agent.imageSynthesis,
+                        this.renderDistortionImage,
+                        "_distortion",
+                        "image_distortion"
+                    );
                 }
                 metadata.thirdPartyCameras = cameraMetadata;
             }

diff --git a/unity/ProjectSettings/ProjectSettings.asset b/unity/ProjectSettings/ProjectSettings.asset
@@ -644,7 +644,7 @@ PlayerSettings:
   webGLThreadsSupport: 0
   webGLDecompressionFallback: 0
   scriptingDefineSymbols:
-    1: 
+    1: UNITY_POST_PROCESSING_STACK_V2
     4: CROSS_PLATFORM_INPUT;MOBILE_INPUT;UNITY_POST_PROCESSING_STACK_V2
     7: CROSS_PLATFORM_INPUT;MOBILE_INPUT;UNITY_POST_PROCESSING_STACK_V2
     13: UNITY_POST_PROCESSING_STACK_V2