Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Object Detection Yolov8 Support #306

Open
Zephyr-S opened this issue Sep 18, 2023 · 1 comment
Open

Object Detection Yolov8 Support #306

Zephyr-S opened this issue Sep 18, 2023 · 1 comment
Labels
hackathon For hachathon/tests

Comments

@Zephyr-S
Copy link

I'm trying to replace the ssd_mobilenet_v2_coco object detection model with my YOLOv8 model on the AMR. I have already modified the paths and related files (xml, label, bin). But when running the object detection docker, the model path did not change. The error shows as below:

[setupvars.sh] OpenVINO environment initialized
[INFO] [1694965557.795466727] [i045.object_detection_node]: [ObjectDetection] Getting parameters ...
[INFO] [1694965557.795635151] [i045.object_detection_node]: Set model_path to: /home/eiforamr/workspace/object_detection/src/prebuilt_object_detection/object_detection/share/object_detection/models/ssd_mobilenet_v2_coco/frozen_inference_graph.xml

[INFO] [1694965557.795653094] [i045.object_detection_node]: Set device to: CPU

[INFO] [1694965557.795666811] [i045.object_detection_node]: Publishing to /camera/detected_objects
[INFO] [1694965557.798381064] [i045.object_detection_node]: Publishing to /camera/image_tracked
[INFO] [1694965557.800236433] [i045.object_detection_node]: Publishing to /camera/tracked_objects
[INFO] [1694965557.800437660] [i045.object_detection_node]: Publishing to /object/image_for_remote
[INFO] [1694965557.802707843] [i045.object_detection_node]: Subscribing to /camera/color/image_raw
[INFO] [1694965557.803233690] [i045.object_detection_node]: Subscribing to /camera/aligned_depth_to_color/image_raw
[INFO] [1694965557.803447426] [i045.object_detection_node]: Subscribing to /camera/aligned_depth_to_color/camera_info
[INFO] [1694965557.803859974] [i045.object_detection_node]: Subscribing to /object/remote_results
[INFO] [1694965557.804077811] [i045.object_detection_node]: Subscribing to /object/enable_remote_inference
InferenceEngine: 0x7fab884aa030
Loading Inference Engine
Loading network files
Path to model: /home/eiforamr/workspace/object_detection/src/prebuilt_object_detection/object_detection/share/object_detection/models/ssd_mobilenet_v2_coco
Model file name: frozen_inference_graph.xml
Base file name: frozen_inference_graph
[ INFO ] Loading Inference Engine
[ INFO ] Device info: 
[ INFO ] 	CPU
	MKLDNNPlugin version ......... 2021.4
	Build ........... 2
Loading network files
[ ERROR ] Unknown model format! Cannot find reader for model format: xml and read the model: /home/eiforamr/workspace/object_detection/src/prebuilt_object_detection/object_detection/share/object_detection/models/ssd_mobilenet_v2_coco/frozen_inference_graph.xml. Please check that reader library exists in your PATH.
[INFO] [1694965557.845248517] [i045.object_detection_node]: Start Tracking Bounding Boxes
[WARN] [1694965558.880644127] [i045.object_detection_node]: No data received in CameraTopic instance

I tried to search the "ssd_mobilenet_v2" in the folder and the result is:

$ grep -r -n ssd_mobilenet workspace
Binary file workspace/object_detection/src/prebuilt_object_detection/object_detection/bin/object_detection_node matches
Binary file workspace/object_detection/src/prebuilt_object_detection/object_detection

I can find the model_path in the two files:
image
How can I compile my own Yolov8 model and update the model path?

@LewisLiuPub
Copy link
Contributor

This issue happened 2 errors:

  • [Error-1] You probably didn't set mode_type as yolov8 in the config file, please see this config file as reference.
  • [Error-2] The implementation in the repo doesn't match the latest ultralytics pretrained yolov8.2 model, which should be fixed.

So this issue is still openning for Error-2: Update related code to support latest ultralytics models:

  • Yolov8 ( Detection & Segmentation)
  • Yolov10 (Detection)
  • Others (Optional)

@LewisLiuPub LewisLiuPub added the hackathon For hachathon/tests label Jul 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
hackathon For hachathon/tests
Projects
None yet
Development

No branches or pull requests

2 participants