custom_model_postproc

Description

Custom torch modules for specific torch models.

Sub-modules

efficientdet_extension

Custom modules for handling and modifying the EfficientDet models from https://github.com/rwightman/efficientdet-pytorch.

mask_rcnn_extensions

Custom modules for handling and modifying the pytorch Mask R-CNN model.

models_common

Common utility functions for custom model extensions.