Comments (3)
you have to exclude the detect layer while exporting from pytorch to onnx, you are trying to export end-to-end model.
from ncnn-android-yolov7.
Hi xiang-wu,
Thanks for your quick feedback. I've corrected my error, but I'm still having problems with the model.
After exporting the custom weights to nccn and running the application, I realized that net doesn't do any detection (the anchor boxes are never shown), the application works like a camera.
Below is the step by step that I am using.
a) My network is used to detect fruit diseases. It comprises about 20,000 images, with 9 classes and achieves a mAP of approximately 87% in yolov7-tiny
b) .PT --> onnx
python export.py --weights yolo-papaya-tiny.pt --simplify --topk-all 100 --iou-thres 0.50 --conf-thres 0.15 --img-size 448 448 --max-wh 448
c) onnx --> ncnn
./onnx2ncnn yolo-papaya-tiny.onnx yolo-papaya-tiny.param yolo-papaya-tiny.bin
d) After conversion, the weights are renamed and transferred to the application folder
If you have the time and inclination to point me out where I'm going wrong, I'd be very grateful.
I am also attaching the weights after converting to ncnn
ncnn - yolo-papaya-tiny.zip
Origin-yolo-papaya-tiny.zip
from ncnn-android-yolov7.
you have to exclude the detect layer while exporting from pytorch to onnx, you are trying to export end-to-end model.
能加个邮箱吗?我们有些问题想要咨询,我的邮箱在主页介绍那里
from ncnn-android-yolov7.
Related Issues (19)
- direct detect interface HOT 1
- Is there a way to add YOLOv6 to the app to compare?
- cmake build error HOT 1
- How to use yolov7.pt & yolov7-w6-pose.pt HOT 1
- How to switch camera used? HOT 1
- "Unsupported Java." and Using Custom Model Successfully HOT 6
- 老哥可以出个yolov5的分类模型推理吗? HOT 1
- 老师麻烦看下一直报这个错是啥问题
- Build error `clang++: error: unknown argument: '-static-openmp'`
- pt->onnx->ncnn HOT 2
- use the deform_conv in yolov7, and use pnnx get a ncnn file, but crush on the phone
- onnx2ncnn.exe problem HOT 14
- about fps HOT 1
- cmake build error HOT 1
- camera resolution
- problem using custom weights HOT 3
- Video showing but no object detections HOT 9
- fp16 model problem
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from ncnn-android-yolov7.