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
1