[YOLOv5s] Array out of bounds when using Live Camera

[baobaoboa]

I'm using YOLOv5s weights since I want to test it out first but there's a problem in LIVE Camera, Anyone know how to fix this?

ERROR:

E/AndroidRuntime: FATAL EXCEPTION: ModuleActivity
    Process: org.pytorch.demo.objectdetection, PID: 30918
    java.lang.ArrayIndexOutOfBoundsException: length=151200; index=151200
        at org.pytorch.demo.objectdetection.PrePostProcessor.outputsToNMSPredictions(PrePostProcessor.java:136)
        at org.pytorch.demo.objectdetection.ObjectDetectionActivity.analyzeImage(ObjectDetectionActivity.java:111)
        at org.pytorch.demo.objectdetection.ObjectDetectionActivity.analyzeImage(ObjectDetectionActivity.java:29)
        at org.pytorch.demo.objectdetection.AbstractCameraXActivity.lambda$setupCameraX$2$AbstractCameraXActivity(AbstractCameraXActivity.java:90)
        at org.pytorch.demo.objectdetection.-$$Lambda$AbstractCameraXActivity$RN1q24ueM_5tBGCPi8IvvsgIhSE.analyze(Unknown Source:2)
        at androidx.camera.core.ImageAnalysisAbstractAnalyzer.analyzeImage(ImageAnalysisAbstractAnalyzer.java:57)
        at androidx.camera.core.ImageAnalysisNonBlockingAnalyzer$1.run(ImageAnalysisNonBlockingAnalyzer.java:135)
        at android.os.Handler.handleCallback(Handler.java:938)
        at android.os.Handler.dispatchMessage(Handler.java:99)
        at android.os.Looper.loop(Looper.java:236)
        at android.os.HandlerThread.run(HandlerThread.java:67)
I/Process: Sending signal. PID: 30918 SIG: 9