tfm.core.exp_factory.get_exp_config

Looks up the ExperimentConfig according to the exp_name.

tfm.core.exp_factory.get_exp_config(
    exp_name: str
) -> tfm.core.base_trainer.ExperimentConfig

Allowed values for exp_name:

exp_nameDescription
cascadercnn_spinenet_cocoCOCO object detection with Cascade RCNN-RS with SpineNet backbone.
deit_imagenet_pretrainImage classification on imagenet with vision transformer.
fasterrcnn_resnetfpn_cocoCOCO object detection with Faster R-CNN.
image_classificationImage classification general.
maskrcnn_mobilenet_cocoCOCO object detection with Mask R-CNN with MobileNet backbone.
maskrcnn_resnetfpn_cocoCOCO object detection with Mask R-CNN.
maskrcnn_spinenet_cocoCOCO object detection with Mask R-CNN with SpineNet backbone.
mnv2_deeplabv3_cityscapesImage segmentation on cityscapes with mobilenetv2 deeplabv3.
mnv2_deeplabv3_pascalImage segmentation on pascal with mobilenetv2 deeplabv3.
mnv2_deeplabv3plus_cityscapesImage segmentation on cityscapes with mobilenetv2 deeplabv3plus.
mobilenet_imagenetImage classification on imagenet with mobilenet.
resnet_imagenetImage classification on imagenet with resnet.
resnet_rs_imagenetImage classification on imagenet with resnet-rs.
retinanetRetinaNet general config.
retinanet_mobile_cocoCOCO object detection with mobile RetinaNet.
retinanet_resnetfpn_cocoCOCO object detection with RetinaNet.
retinanet_spinenet_cocoCOCO object detection with RetinaNet using SpineNet backbone.
revnet_imagenetReturns a revnet config for image classification on imagenet.
seg_deeplabv3_pascalImage segmentation on pascal voc with resnet deeplabv3.
seg_deeplabv3plus_cityscapesImage segmentation on cityscapes with resnet deeplabv3+.
seg_deeplabv3plus_pascalImage segmentation on pascal voc with resnet deeplabv3+.
seg_resnetfpn_pascalImage segmentation on pascal voc with resnet-fpn.
semantic_segmentationSemantic segmentation general.
video_classificationVideo classification general.
video_classification_kinetics400Video classification on Kinetics 400 with resnet.
video_classification_kinetics600Video classification on Kinetics 600 with resnet.
video_classification_kinetics700Video classification on Kinetics 700 with resnet.
video_classification_kinetics700_2020Video classification on Kinetics 700 2020 with resnet.
video_classification_ucf101Video classification on UCF-101 with resnet.
vit_imagenet_finetuneImage classification on imagenet with vision transformer.
vit_imagenet_pretrainImage classification on imagenet with vision transformer.