wuyang0329 / unet

this is a simple demo for image segmentation.----unet网络进行语义分割的demo,用的数据集是KITTI
97 stars 38 forks source link

请教权重是如何打印出来的? #1

Open jizhang02 opened 5 years ago

jizhang02 commented 5 years ago

image 您好,请问这部分的权重输出是如何操作的呢?我没有找到对应的代码。谢谢。 conv2d_24中的数字24代表什么含义呢?

wuyang0329 commented 5 years ago

image 您好,请问这部分的权重输出是如何操作的呢?我没有找到对应的代码。谢谢。 conv2d_24中的数字24代表什么含义呢? 这个不是权重输出呢,这个是网络结构的输出,包括了每一层的参数数量,用的是summary函数,对应的位置是model文件的第62行 model.summary() conv2d_24中的24代表这个是第24个卷积层,这个是卷积层的默认名称,你可以给每一层定义自己熟悉的名称

jizhang02 commented 5 years ago

感谢回答! 还有一个问题不太明白。就是测试的时候,输入的图像是1242375,但是输出的图像是256256. 而且训练时的标签是512*512,这样的话, 它的损失函数是如何计算的呢?谢谢