efficientdet_extension

Description

Custom modules for handling and modifying the EfficientDet models from https://github.com/rwightman/efficientdet-pytorch. Make sure to install the corresponding package via

# EfficientDet pytorch implementation
pip install -e "git+https://github.com/rwightman/efficientdet-pytorch.git@75e16c2f#egg=effdet"

Classes

EfficientDetToSegMask

Turn batch output of torchvision Mask R-CNN into segmentation masks for given label.