Converting TAO-trained MaskRCNN models to ONNX for CPU inference

Accelerated Computing Intelligent Video Analytics TAO Toolkit
, ,
4

Using TAO 5.3.0, upon running the relevant command (10. Deploy!):

!tao model mask_rcnn export -m $USER_EXPERIMENT_DIR/experiment_dir_retrain/model.epoch-$NUM_EPOCH.tlt \
                      -e $SPECS_DIR/maskrcnn_retrain_resnet50.txt \
                      -k tlt_encode \
                      --gen_ds_config

I get an UFF file instead of an ONNX file. Specifically, the logs show:

⋮ 
2024-10-02 04:12:25,307 [TAO Toolkit] [INFO] nvidia_tao_tf1.cv.mask_rcnn.export.exporter 300: Converted model was saved into /usr/src/app/mask_rcnn/experiment_dir_unpruned/model.epoch-199.uff
⋮

Is this expected?