How to make a custom YOLO package

asked 2018-08-17 00:15:00 -0600

Nelle
21 ●10 ●12 ●13

I am very new to ROS. I have been able to make publishers and subscribers now. And also some custom messages. My problem is that we are a team of people making a rescue robot that will detect body parts and faces. One team is already making the customized YOLO and our team is focused on the ROS part and harware interfacing. Now my problem is that we have to turn the customized YOLO into a ROS package so that we can use it in ROS and view the images in the camera thru the RVIZ in a remote computer. I'm running on kinetic and raspbian, btw. The camera would be in the rpi.

How do we do this?

Comments

answered 2018-08-17 02:03:13 -0600

Choco93
685 ●9 ●37 ●24

It's just like using yolo normally with bunch of saved images or your webcam. You subscribe to your image topic which will give you ROS Image, you convert that image into bgr (cv format) conversion.

Then you just feed that converted image to yolo and let it do it's magic. Using opencv you can also draw the bounding boxes onto the image and republish it as a ROS Image.

edit flag offensive delete link

Comments

I have already made a raspberry pi camera show images thru rviz. I have this neat node package. How can I launch yolo thru ROS if it isn't a package? Really confused here. And one more thing, I'm not really familiar with using yolo.

Nelle ( 2018-08-17 02:12:58 -0600 )edit

You have to make yolo a ROS package, I don't think there is another solution to that.

Choco93 ( 2018-08-17 02:18:30 -0600 )edit

How to do that exactly? Any tutorials and what nots you can share?

Nelle ( 2018-08-17 02:38:41 -0600 )edit

Unless I'm misunderstanding something, there appears to be no reason to create yet another ROS YOLO pkg, as there are quite a few available already. I'd really advise to take a look at those, as they potentially save you a lot of work.

gvdhoorn ( 2018-08-17 02:40:30 -0600 )edit

As mentioned in the question that they are customizing yolo, I am assuming they are making some specific changes for their case which I am not sure how easy can be to add to an existing package. If only making their own trained model then it should be fairly easy to use existing packages.

Choco93 ( 2018-08-17 02:46:29 -0600 )edit

The OP will have to clarify this. I've seen people claim they are creating a 'custom deep learning system', but in the end were only loading an alternative model.

Even if the OP really needs a different network or some other customisation, starting with existing pkgs is probably a good thing.

gvdhoorn ( 2018-08-17 02:55:29 -0600 )edit

I'm not really sure how the other team is "customizing" YOLO. But if we are to add only trained models, how can we add this to an existing ros yolo like this one?

Nelle ( 2018-08-17 02:57:04 -0600 )edit

I'm not really sure how the other team is "customizing" YOLO

then personally I would first try to get some info from them. Seeing as your task appears to be to integrate the results of the 'other team', wouldn't it be good to know what they are actually doing?

gvdhoorn ( 2018-08-17 03:02:21 -0600 )edit

add a comment

How to make a custom YOLO package

Comments

1 Answer

Comments

Question Tools

Stats

Related questions

How to make a custom YOLO package edit

Comments

1 Answer

Comments

Question Tools

Stats

Related questions

How to make a custom YOLO package