bd6abd23feba4f7955a6f240412e7be06c5b6ca3,tests/unittests/test_utils_export.py,,test_export_model_zoo,#,9

Before Change

    for model in pretrained_model_list():
        if "i3d" in model: continue    // 3D model do not support it now, skip. Data shape is 5D.
        print("exporting:", model)
        kwargs = {"data_shape":(480, 480, 3)} if "deeplab" in model or "psp" in model else {}
        if "_gn" in model:
            continue
        try:
            gcv.utils.export_block(model, gcv.model_zoo.get_model(model, pretrained=True), **kwargs)

After Change

    for model in pretrained_model_list():
        print("exporting:", model)

        if "deeplab" in model or "psp" in model:
            // semantic segmentation models require fixed data shape
            kwargs = {"data_shape":(480, 480, 3)}
        elif "3d" in model:
            // video action recognition models require 4d data shape
            kwargs = {"data_shape":(3, 32, 224, 224), "layout":"CTHW", "preprocess":None}
        elif "slowfast_4x16" in model:
            // video action recognition models require 4d data shape
            kwargs = {"data_shape":(3, 36, 224, 224), "layout":"CTHW", "preprocess":None}
        elif "slowfast_8x8" in model:

Instances