search

pytorch / android-demo-app

PyTorch android examples of usage in applications
1.48k stars 607 forks source link

value of mOutputRow for YOLOv7 model in objectdetection/PrePostProcessor.java ? #301

Open dsbyprateekg opened 1 year ago

dsbyprateekg commented 1 year ago

Hi,

I am trying to utilize android-demo-app/ObjectDetection/app/src/main/java/org/pytorch/demo/objectdetection/PrePostProcessor.java but with YOLOv7 model instead of YOLOv5.

Can anybody please tell me what should be the value here instead of 25200 for YOLOv7 model- image

NeighborhoodCoding commented 1 year ago

YOLOv3 is better...