def test_export_model_zoo():
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:
# video action recognition models require 4d data shape
kwargs = {'data_shape':(3, 40, 224, 224), 'layout':'CTHW', 'preprocess':None}
else:
kwargs = {}
if '_gn' in model:
continue
try:
gcv.utils.export_block(model, gcv.model_zoo.get_model(model, pretrained=True), **kwargs)
except ValueError:
# ignore non defined model name
pass
except AttributeError:
# deeplab model do not support it now, skip
pass
def test_export_model_zoo():
for model in pretrained_model_list():
print('exporting:', model)
try:
gcv.utils.export_block(model, gcv.model_zoo.get_model(model, pretrained=True))
except:
# ignore non defined model name
pass
def test_export_model_zoo():
for model in pretrained_model_list():
print('exporting:', model)
try:
gcv.utils.export_block(
model, gcv.model_zoo.get_model(model, pretrained=True))
except:
# ignore non defined model name
pass
def test_export_model_zoo():
for model in pretrained_model_list():
print('exporting:', model)
try:
gcv.utils.export_block(
model, gcv.model_zoo.get_model(model, pretrained=True))
except ValueError:
# ignore non defined model name
pass
except AttributeError:
# deeplab model do not support it now, skip
pass
def test_export_model_zoo():
for model in pretrained_model_list():
print('exporting:', model)
kwargs = {'data_shape': (480, 480, 3)} if 'deeplab' in model else {}
try:
gcv.utils.export_block(
model, gcv.model_zoo.get_model(model, pretrained=True),
**kwargs)
except ValueError:
# ignore non defined model name
pass
except AttributeError:
# deeplab model do not support it now, skip
pass
