NVIDIA / semantic-segmentation

Nvidia Semantic Segmentation monorepo
BSD 3-Clause "New" or "Revised" License
1.78k stars 388 forks source link

Use this model for background removal. #74

Open kashyappiyush1998 opened 4 years ago

kashyappiyush1998 commented 4 years ago

How can we use this model for background removal from image or video?

karansapra commented 4 years ago

Sorry but can you elaborate what you are trying to do. Our model is only for semantic segmentation.

kashyappiyush1998 commented 4 years ago

Sorry, for the very late question. But what I want to do is create a mask for background removal from images, can I use this model for the segmentation of 2 objects from the image, foreground, and background.

karansapra commented 4 years ago

Yes, you can. But you will have to create a new dataloader, you can look at the mapillary dataset to see how to create a new dataloader file and dataset/init.py to see how to integrate it.