https://drive.google.com/drive/folders/1KzBjmCOG9ghcq9L6-iqfz6QwBQq6Hl4_?usp=sharing or https://share.weiyun.com/td9CRDhW open pytorch_onnx_tensorrt_unet\code\unet_to_onnx.py

pip install torch