imagenet-swav-resnet50w2 / convert_onnx_2_tf.sh
lixc
SwAV ResNet50w2 of Layer2
b96c6e9
raw
history blame
379 Bytes
#!/bin/bash
# cp /opt/software/github/swav/traced_swav_imagenet_layer2.pt ./
onnxsim /opt/software/github/swav/swav_imagenet_layer2.onnx swav_imagenet_layer2_sim.onnx
# https://github.com/onnx/onnx-tensorflow
onnx-tf convert --infile swav_imagenet_layer2_sim.onnx --outdir tf_saved_model
tensorflowjs_converter --input_format tf_saved_model tf_saved_model tfjs_saved_model/