Comments (4)
@arkerman hello! Thank you for providing detailed information about your issue. It seems like there might be a discrepancy in how the class labels are being handled between the .pt
model and the .engine
model.
One possible cause could be related to the class labels not being correctly integrated or recognized in the conversion process to the TensorRT engine format. Here are a couple of steps you can take to troubleshoot this issue:
-
Verify Class Labels in YAML: Ensure that the
mydata.yaml
file used during the training and the export process has the correct class labels listed and in the same order. -
Check Export Logs: Review the logs generated during the export process for any warnings or errors related to class labels or the conversion process.
-
Re-export the Model: Try re-exporting the
.pt
model to the.engine
format, ensuring that all command line parameters are correctly specified and consistent with the training phase.
If these steps do not resolve the issue, please provide the contents of your mydata.yaml
and any relevant logs from the export process, so we can help diagnose the problem further.
Thank you for using YOLOv5, and we appreciate your contribution to making the tool better! 🚀
from yolov5.
@glenn-jocher Hi! Really appreciate your suggestions! I debuged the predict.py code, and found it what cause the question.
I forgot to add the argument --data
.
I used the command line below, and the everything is ok now.
python segment/predict.py --weights runs/train-seg/exp13/weights/best.engine --source E:\data\mydata\test\images --data mydata.yaml
Thanks again!
from yolov5.
@arkerman That's great to hear! I'm glad you were able to resolve the issue by including the --data
argument. If you have any more questions or run into further issues, feel free to reach out. Happy detecting! 🚀
from yolov5.
👋 Hello there! We wanted to give you a friendly reminder that this issue has not had any recent activity and may be closed soon, but don't worry - you can always reopen it if needed. If you still have any questions or concerns, please feel free to let us know how we can help.
For additional resources and information, please see the links below:
- Docs: https://docs.ultralytics.com
- HUB: https://hub.ultralytics.com
- Community: https://community.ultralytics.com
Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!
Thank you for your contributions to YOLO 🚀 and Vision AI ⭐
from yolov5.
Related Issues (20)
- Syntax and understanding questions about reading tensorflow lite results HOT 1
- A Error which blast my mind.... HOT 1
- Yolov5 Int8 export in PyTorch HOT 10
- Video inference with YOLOv5 model in python HOT 3
- How to disable or add new scales of prediction? HOT 5
- Installation on Windows 7 32 bits HOT 3
- Artificial Neural Network - interpreting model.save output HOT 1
- Origin of warmup_bias_lr? HOT 3
- Silicon Mac GPU Support for training HOT 1
- Split features map of data HOT 1
- incorrect detections for cars after fine-tuning yolov5l HOT 1
- tried to run yolov5 "detect.py" with pretrained model yolov8x.pt and xView.yaml HOT 1
- Low disk space causes memory leak HOT 3
- Error with fbgemm.dll file when using Torch! HOT 5
- Yolov5 inventing label on validation set HOT 3
- challenges faced in Xtream1 tool while creating class ID's HOT 1
- Facing issues while changing class ID values HOT 1
- background color of image and other causes? HOT 1
- ModuleNotFoundError: No module named 'models.yolo'. HOT 3
- RuntimeError: The size of tensor a (80) must match the size of tensor b (60) at non-singleton dimension 3 HOT 4
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 yolov5.