Update InstanceSegmentationLabeler.cs

Mohsen Kamalzadeh · Mohsen Kamalzadeh · commit a3ea2288047d · 2021-02-01T16:42:50.000-08:00
diff --git a/com.unity.perception/Runtime/GroundTruth/Labelers/InstanceSegmentationLabeler.cs b/com.unity.perception/Runtime/GroundTruth/Labelers/InstanceSegmentationLabeler.cs
@@ -117,14 +117,17 @@ void OnRenderedObjectInfosCalculated(int frame, NativeArray<RenderedObjectInfo>
             }
         }
 
-        void OnImageCaptured(int frame, NativeArray<Color32> data, RenderTexture renderTexture)
+        void OnImageCaptured(int frameCount, NativeArray<Color32> data, RenderTexture renderTexture)
         {
+            if (!m_AsyncAnnotations.ContainsKey(frameCount))
+                return;
+
             using (s_OnImageReceivedCallback.Auto())
             {
                 m_CurrentTexture = renderTexture;
 
-                m_InstancePath = $"{k_Directory}/{k_FilePrefix}{frame}.png";
-                var localPath = $"{Manager.Instance.GetDirectoryFor(k_Directory)}/{k_FilePrefix}{frame}.png";
+                m_InstancePath = $"{k_Directory}/{k_FilePrefix}{frameCount}.png";
+                var localPath = $"{Manager.Instance.GetDirectoryFor(k_Directory)}/{k_FilePrefix}{frameCount}.png";
 
                 var colors = new NativeArray<Color32>(data, Allocator.TempJob);
 

Original file line number	Diff line number	Diff line change
`@@ -117,14 +117,17 @@ void OnRenderedObjectInfosCalculated(int frame, NativeArray<RenderedObjectInfo>`
`117`	`117`	`}`
`118`	`118`	`}`
`119`	`119`
`120`		`- void OnImageCaptured(int frame, NativeArray<Color32> data, RenderTexture renderTexture)`
	`120`	`+ void OnImageCaptured(int frameCount, NativeArray<Color32> data, RenderTexture renderTexture)`
`121`	`121`	`{`
	`122`	`+ if (!m_AsyncAnnotations.ContainsKey(frameCount))`
	`123`	`+ return;`
	`124`	`+`
`122`	`125`	`using (s_OnImageReceivedCallback.Auto())`
`123`	`126`	`{`
`124`	`127`	`m_CurrentTexture = renderTexture;`
`125`	`128`
`126`		`- m_InstancePath = $"{k_Directory}/{k_FilePrefix}{frame}.png";`
`127`		`- var localPath = $"{Manager.Instance.GetDirectoryFor(k_Directory)}/{k_FilePrefix}{frame}.png";`
	`129`	`+ m_InstancePath = $"{k_Directory}/{k_FilePrefix}{frameCount}.png";`
	`130`	`+ var localPath = $"{Manager.Instance.GetDirectoryFor(k_Directory)}/{k_FilePrefix}{frameCount}.png";`
`128`	`131`
`129`	`132`	`var colors = new NativeArray<Color32>(data, Allocator.TempJob);`
`130`	`133`