ayoolaolafenwa / PixelLib

Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multiple classes for segmentation

Rock-Shell opened this issue · comments

commented

I was trying to segment multiple but NOT all classes, and found "change_bg_img" function from "pixellib.tune_bg.alter_bg" to be helpful. But in order to perform segmentation for multiple classes, I had to use the function multiple times with different values in the detect parameter.

for reference
def change_bg_img(self, f_image_path,b_image_path, output_image_name = None, verbose = None, detect = None):

I found that detect parameter takes a single value. Instead, if it takes a list of values, the problem could be easily resolved.
Please tell, if there is some better way to approach the problem. If not, I would like to work on this issue.