search

sacmehta / EdgeNets

This repository contains the source code of our work on designing efficient CNNs for computer vision
MIT License
412 stars 82 forks source link

How to do semantic segmentation of two categories ？ #21

Closed lighTQ closed 4 years ago

lighTQ commented 4 years ago

(including background and palm)

I am using a dataset in pascal voc format, so I need to do extra opcodes besides modifying VOC_CLASS_LIST, and loss_type to bce

sacmehta commented 4 years ago

You can edit this custom file as per your need.

https://github.com/sacmehta/EdgeNets/blob/4aa09637ad4bc2337e887c3ff0633e819fb2bcde/data_loader/segmentation/custom_dataset_loader.py#L20