search

yangliu9527 / Object_SLAM

A visual SLAM system with Object-Aware Data Association and Semantically Constrained Optimization
Other
9 stars 0 forks source link

about datasets #3

Closed sumin12345678 closed 3 months ago

sumin12345678 commented 3 months ago

Firstly, thank you for your excellent work. I would like to know how datasets are processed and if there are any tutorials or Python code available. Looking forward to your reply. Thanks

yangliu9527 commented 3 months ago

I used YOLACT to acquire the instance segmentation results of original images. I‘m sorry that the script I used is missing. But I can give you some suggestions based on the source code of YOLACT. The eval.py in its repository is the main code to extract semantic results but it only outputs a visible result instead of detailed data of masks, bounding boxes. So you have to modify eval.py by yourself to get what you want. You can access to the raw results from predispaly(), and you need to modify evalimage(), evalimages() to store results.

By the way, nowadays there are more reliable instance segmentation methods you can try.

sumin12345678 commented 3 months ago

When I used YOLOV8S SEG pre trained model for segmentation, I found that there were not as many objects detected in the same image as YOLACT, but that may be related to model training. Thank you for your prompt and detailed answer. Wishing you a happy life!