文件名称:Unet训练和部署代码
文件大小:3.79MB
文件格式:ZIP
更新时间:2022-08-06 21:17:57
unet libtorch libtorch部署代码
Unet训练和libtorch部署代码: 1、训练部分 2、libtorch部署方法
【文件预览】:
Unet_pytorch_libtorch-master
----get_miou_prediction.py(1KB)
----img()
--------street.jpg(437KB)
----readme.md(5KB)
----json_to_dataset.py(2KB)
----tools()
--------torch2onnx.py(456B)
--------__pycache__()
--------predict.py(5KB)
--------Net_Vision.py(452B)
--------pth2pt.py(330B)
----nets()
--------__pycache__()
--------unet_training.py(5KB)
--------vgg.py(3KB)
--------unet.py(3KB)
----train.py(10KB)
----demo.py(2KB)
----VOCdevkit()
--------voc2unet.py(1KB)
----Libtorch_unet()
--------Unet()
----miou.py(5KB)
----utils()
--------metrics.py(1KB)
--------__pycache__()
--------dataloader.py(7KB)
--------dataloader_medical.py(5KB)
----unet.py(10KB)