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

Multiple images for image segmentation #72

Closed
archit1508 opened this issue Jun 13, 2021 · 1 comment
Closed

Multiple images for image segmentation #72

archit1508 opened this issue Jun 13, 2021 · 1 comment

Comments

@archit1508
Copy link

Hi,

We are young researchers and find your tool very useful for segmentation. Currently we are in the process of using the image segmentation model for fruit quality evaluation. The model pretrained on COCO works.

How can we save the segmented instances using your library for multiple images and with the same names as our original dataset ?

Also please note, we will provide proper citations and contribution mentions in any publications.

Thanks

@ayoolaolafenwa
Copy link
Owner

@archit1508 upgrade to the latest version of PixelLib that supports batch prediction.
upgrade using;

pip3 install pixellib --upgrade

Code for batch prediction

import pixellib
from pixellib.instance import instance_segmentation

ins = instance_segmentation()
ins.load_model("mask_rcnn_coco.h5")
ins.segmentBatch("input_folder",  show_bboxes=True, output_folder_name = "output_folder")

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants