search

zhen8838 / K210_Yolo_framework

Yolo v3 framework base on tensorflow, support multiple models, multiple datasets, any number of output layers, any number of anchors, model prune, and portable model to K210 !
MIT License
248 stars 73 forks source link

输入大小 #50

Open junze6 opened 2 years ago

junze6 commented 2 years ago

输入大小只能是224x320吗，如果输入是224x224该怎么改main.c文件

zhen8838 commented 2 years ago

输入大小是和模型相关的,首先修改模型,然后再把main.c里面的10改成7.

junze6 commented 2 years ago

我尝试使用pytorch框架下yolov3-tiny训练剪枝完转成kmodel，输入224x224,结果检测不出来东西，10改7，20改14，还有一些其他参数改了都没用，会是模型的问题吗

zhen8838 commented 2 years ago

不用剪枝, k210不支持稀疏加速, 这个可能是量化误差问题,但是不至于一点也检测不出来. 你可以参考我写的yolox的例子 , 先检查ncc infer的结果和原始模型推理结果是否一致. 例子里面有个cpu文件夹, 可以方便来做pc端的验证.

junze6 commented 2 years ago

非常感谢您的回答，代码里只有单独的precision,recall指标，请问我想计算map应该怎么做。

zhen8838 commented 2 years ago

写点代码把结果导出来，然后用一个map的库去算。

发自我的iPhone

------------------ 原始邮件 ------------------ 发件人: junze6 @.> 发送时间: 2022年4月16日 21:02 收件人: zhen8838/K210_Yolo_framework @.> 抄送: 郑启航 @.>, Comment @.> 主题: 回复：[zhen8838/K210_Yolo_framework] 输入大小 (Issue #50)

非常感谢您的回答，代码里只有单独的precision,recall指标，请问我想计算map应该怎么做。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

junze6 commented 2 years ago

好的，感谢您的回复，我利用pycocotools,现在已经可以计算每个epoch的map了