feat. adapt to yolo 160

Chen-Yulin · Apr 14, 2024 · 6a2de05 · 6a2de05
1 parent f7f7fec
commit 6a2de05
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Assets/Script/Detection/NN/YOLOv8OutputReader.cs b/Assets/Script/Detection/NN/YOLOv8OutputReader.cs
@@ -12,8 +12,8 @@ public class YOLOv8OutputReader
         public static float DiscardThreshold = 0.1f;
         protected const int ClassesNum = 80;
         const int BoxesPerCell = 525;
-        const int InputWidth = 240;
-        const int InputHeight = 240;
+        const int InputWidth = 160;
+        const int InputHeight = 160;
 
         public IEnumerable<ResultBox> ReadOutput(Tensor output)
         {