View source on GitHub |
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_name | Description |
---|---|
cascadercnn_spinenet_coco | COCO object detection with Cascade RCNN-RS with SpineNet backbone. |
deit_imagenet_pretrain | Image classification on imagenet with vision transformer. |
fasterrcnn_resnetfpn_coco | COCO object detection with Faster R-CNN. |
image_classification | Image classification general. |
maskrcnn_mobilenet_coco | COCO object detection with Mask R-CNN with MobileNet backbone. |
maskrcnn_resnetfpn_coco | COCO object detection with Mask R-CNN. |
maskrcnn_spinenet_coco | COCO object detection with Mask R-CNN with SpineNet backbone. |
mnv2_deeplabv3_cityscapes | Image segmentation on cityscapes with mobilenetv2 deeplabv3. |
mnv2_deeplabv3_pascal | Image segmentation on pascal with mobilenetv2 deeplabv3. |
mnv2_deeplabv3plus_cityscapes | Image segmentation on cityscapes with mobilenetv2 deeplabv3plus. |
mobilenet_imagenet | Image classification on imagenet with mobilenet. |
resnet_imagenet | Image classification on imagenet with resnet. |
resnet_rs_imagenet | Image classification on imagenet with resnet-rs. |
retinanet | RetinaNet general config. |
retinanet_mobile_coco | COCO object detection with mobile RetinaNet. |
retinanet_resnetfpn_coco | COCO object detection with RetinaNet. |
retinanet_spinenet_coco | COCO object detection with RetinaNet using SpineNet backbone. |
revnet_imagenet | Returns a revnet config for image classification on imagenet. |
seg_deeplabv3_pascal | Image segmentation on pascal voc with resnet deeplabv3. |
seg_deeplabv3plus_cityscapes | Image segmentation on cityscapes with resnet deeplabv3+. |
seg_deeplabv3plus_pascal | Image segmentation on pascal voc with resnet deeplabv3+. |
seg_resnetfpn_pascal | Image segmentation on pascal voc with resnet-fpn. |
semantic_segmentation | Semantic segmentation general. |
video_classification | Video classification general. |
video_classification_kinetics400 | Video classification on Kinetics 400 with resnet. |
video_classification_kinetics600 | Video classification on Kinetics 600 with resnet. |
video_classification_kinetics700 | Video classification on Kinetics 700 with resnet. |
video_classification_kinetics700_2020 | Video classification on Kinetics 700 2020 with resnet. |
video_classification_ucf101 | Video classification on UCF-101 with resnet. |
vit_imagenet_finetune | Image classification on imagenet with vision transformer. |
vit_imagenet_pretrain | Image classification on imagenet with vision transformer. |