Merge pull request #39 from Project-MONAI/anishag14/restructure-scripts

Move the helper script out of src/

scripts/README.md

@@ -0,0 +1,7 @@
+This folder contains helper scripts.
+
+## convert_pytorch2onnx.py
+
+This script will take the path of the Pytorch model and input tensor size. To
+run the script ..code: `python scripts/convert_pytorch2onnx.py` to generate
+the ONNX model.

src/utils/convert_pytorch2onnx.py → scripts/convert_pytorch2onnx.py

@@ -13,23 +13,24 @@
 
 import torch.onnx
 
-model_path = "model_jit.pt"  # modify with your own path to the pytorch model
-
-
+# modify model_path with the path to the pytorch model
+model_path = "model_jit.pt"
 torch_model = torch.jit.load(model_path)
 torch_model = torch_model.eval()
 
+# provide the input torch size that pytorch model takes
 x = torch.randn(1, 3, 1024, 1264)
 torch_out = torch_model(x)
 input_names = ["INPUT__0"]
 output_names = ["OUTPUT__0"]
+
 # Export the model
 torch.onnx.export(
     torch_model,  # model being run
     x,  # model input (or a tuple for multiple inputs)
     "monai_unet_pyt.onnx",  # where to save the model (can be a file or file-like object)
     example_outputs=torch_out,
-    # export_params=True,        # store the trained parameter weights inside the model file
+    # export_params=True,  # store the trained parameter weights inside the model file
     verbose=True,
     input_names=input_names,
     output_names=output_names,