pytorch-unet:用于图像分割的U-NetFullyConvNet(FCN)的简单PyTorch实现

时间:2024-05-31 07:04:24
【文件属性】:

文件名称:pytorch-unet:用于图像分割的U-NetFullyConvNet(FCN)的简单PyTorch实现

文件大小:160KB

文件格式:ZIP

更新时间:2024-05-31 07:04:24

image-segmentation unet semantic-segmentation fully-convolutional-networks 附件源码

UNet / FCN PyTorch 该存储库包含U-Net和FCN的简单PyTorch实现,这是Ronneberger等人提出的深度学习细分方法。 和龙等。 用于训练的合成图像/遮罩 首先克隆存储库并cd到项目目录。 import matplotlib . pyplot as plt import numpy as np import helper import simulation # Generate some random images input_images , target_masks = simulation . generate_random_data ( 192 , 192 , count = 3 ) for x in [ input_images , target_masks ]: print ( x . shape ) print ( x . m


【文件预览】:
pytorch-unet-master
----.gitignore(1KB)
----pytorch_resnet18_unet.ipynb(81KB)
----images()
--------output_9_1.png(14KB)
--------output_2_2.png(4KB)
--------output_0_1.png(12KB)
----pytorch_fcn.ipynb(35KB)
----LICENSE(1KB)
----helper.py(1KB)
----pytorch_unet.py(2KB)
----pytorch_unet_resnet18_colab.ipynb(115KB)
----README.md(26KB)
----simulation.py(3KB)
----pytorch_unet.ipynb(65KB)
----loss.py(354B)

网友评论