Add files via upload

UnitySDK/Assets/ML-Agents/Examples/Crawler/Scripts/CrawlerAgent.cs

     bool isNewDecisionStep;
     int currentDecisionStep;
 
+    private Transform workingTransform;
+
     public override void InitializeAgent()
     {
         jdController = GetComponent<JointDriveController>();
         jdController.SetupBodyPart(leg2Lower);
         jdController.SetupBodyPart(leg3Upper);
         jdController.SetupBodyPart(leg3Lower);
+
+        workingTransform = new GameObject().transform;
     }
 
     /// <summary>
     {
         var rb = bp.rb;
         AddVectorObs(bp.groundContact.touchingGround ? 1 : 0); // Whether the bp touching the ground
-        AddVectorObs(rb.velocity);
-        AddVectorObs(rb.angularVelocity);
+
+        AddVectorObs(workingTransform.InverseTransformVector(rb.velocity));
+        AddVectorObs(workingTransform.InverseTransformDirection(rb.angularVelocity));
 
         if (bp.rb.transform != body)
         {
     {
         jdController.GetCurrentJointForces();
         // Normalize dir vector to help generalize
-        AddVectorObs(dirToTarget.normalized);
+        workingTransform.rotation = Quaternion.LookRotation(dirToTarget);
-        AddVectorObs(body.transform.position.y);
-        AddVectorObs(body.forward);
-        AddVectorObs(body.up);
+        RaycastHit hit;
+        if (Physics.Raycast(body.position, Vector3.down, out hit, 10.0f))
+        {
+            AddVectorObs(hit.distance);
+        }
+        else
+            AddVectorObs(10.0f); 
+        AddVectorObs(workingTransform.InverseTransformVector(body.forward));
+        AddVectorObs(workingTransform.InverseTransformVector(body.up));
         foreach (var bodyPart in jdController.bodyPartsDict.Values)
         {
             CollectObservationBodyPart(bodyPart);
         {
             transform.rotation = Quaternion.LookRotation(dirToTarget);
         }
+        transform.Rotate(Vector3.up,Random.Range(0.0f, 360.0f));
 
         foreach (var bodyPart in jdController.bodyPartsDict.Values)
         {