fix bug

metadriverse · Mar 26, 2024 · 4006c44 · 4006c44
1 parent 28dddc7
commit 4006c44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/metadrive/component/sensors/semantic_camera.py b/metadrive/component/sensors/semantic_camera.py
@@ -38,7 +38,7 @@ def _setup_effect(self):
                 )
             else:
 
-                if label == "PEDESTRIAN":
+                if label == Semantics.PEDESTRIAN.label:
                     # PZH: This is a workaround fix to make pedestrians animated.
                     cam.setTagState(
                         label,
@@ -55,7 +55,7 @@ def _setup_effect(self):
                     cam.setTagState(
                         label,
                         RenderState.make(
-                            # ShaderAttrib.makeOff(),
+                            ShaderAttrib.makeOff(),
                             LightAttrib.makeAllOff(),
                             TextureAttrib.makeOff(),
                             ColorAttrib.makeFlat((c[0] / 255, c[1] / 255, c[2] / 255, 1)),