Describe the solution you'd likeMandatory Inputs:url: URL pointing to a dataset
dataset_name: name of a dataset
annotation_type: select coco or voc annotation
Optional Inputs: By default, we choose EfficientDet as model and tf_efficientdet_lite0 as a backbone
model: efficientdet, faster_rcnn or mask_rcnn. By default, we choose efficientdetbackbone: Corresponding backbone. For more details, check EfficientDet and Faster/Mask RCNN
backbone lists
If the model is efficientdet, backbone = "tf_efficientdet_lite0"
If the model is faster_rcnn or mask_rcnn, the default backbone is:
backbone = backbones.resnet_fpn.resnet50(pretrained=True)
🚀 Feature
Is your feature request related to a problem? Please describe. Create a script that generates an an end-to-end training code snippet:
method signature:
generate_e2e_training(url: str, dataset_name: str, annotation_type="voc", model="efficientdet", backbone=None)
Describe the solution you'd like Mandatory Inputs:
url
: URL pointing to a datasetdataset_name
: name of a datasetannotation_type
: selectcoco
orvoc
annotationOptional Inputs: By default, we choose
EfficientDet
as model andtf_efficientdet_lite0
as a backbonemodel
:efficientdet
,faster_rcnn
ormask_rcnn
. By default, we chooseefficientdet
backbone
: Corresponding backbone. For more details, check EfficientDet and Faster/Mask RCNN backbone listsIf the model is
efficientdet
, backbone = "tf_efficientdet_lite0" If the model isfaster_rcnn
ormask_rcnn
, the default backbone is:backbone = backbones.resnet_fpn.resnet50(pretrained=True)
For
efficientdet
, check out the quickstart example Forfaster_rcnn
, check out the getting_started example Forfaster_rcnn
, check out the mask_rcnn_pennfudan exampleOutput: A ready to train end-to-end training code.
Example: My Awesome Dataset
Inputs:
Output: