-
大小: 10KB文件類型: .zip金幣: 1下載: 0 次發(fā)布日期: 2021-01-09
- 語言: 其他
- 標(biāo)簽: 深度學(xué)習(xí)??
資源簡介
基于Pytorch的UNet、UNet++的代碼.參考csdn的鏈接:https://blog.csdn.net/weixin_40519315/article/details/104581992
代碼片段和文件信息
#?-*-?coding:?utf-8?-*-
import?numpy?as?np
from?torch?import?nn
from?torch.nn?import?functional?as?F
import?torch
from?torchvision?import?models
import?torchvision
class?VGGBlock(nn.Module):
????def?__init__(self?in_channels?middle_channels?out_channels?act_func=nn.ReLU(inplace=True)):
????????super(VGGBlock?self).__init__()
????????self.act_func?=?act_func
????????self.conv1?=?nn.Conv2d(in_channels?middle_channels?3?padding=1)
????????self.bn1?=?nn.BatchNorm2d(middle_channels)
????????self.conv2?=?nn.Conv2d(middle_channels?out_channels?3?padding=1)
????????self.bn2?=?nn.BatchNorm2d(out_channels)
????def?forward(self?x):
????????out?=?self.conv1(x)
????????out?=?self.bn1(out)
????????out?=?self.act_func(out)
????????out?=?self.conv2(out)
????????out?=?self.bn2(out)
???????
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2020-03-04?16:26??NestedUNet_Pytorch\
?????文件????????5710??2019-03-11?21:11??NestedUNet_Pytorch\archs.py
?????文件????????1309??2020-03-04?16:10??NestedUNet_Pytorch\dataset.py
?????目錄???????????0??2020-03-04?16:25??NestedUNet_Pytorch\input\
?????文件???????????0??2019-03-11?21:11??NestedUNet_Pytorch\input\.gitkeep
?????文件????????1047??2019-03-11?21:11??NestedUNet_Pytorch\losses.py
?????文件????????3999??2019-03-11?21:11??NestedUNet_Pytorch\metrics.py
?????目錄???????????0??2020-03-04?16:25??NestedUNet_Pytorch\models\
?????文件???????????0??2019-03-11?21:11??NestedUNet_Pytorch\models\.gitkeep
?????目錄???????????0??2020-03-04?16:25??NestedUNet_Pytorch\output\
?????文件???????????0??2019-03-11?21:11??NestedUNet_Pytorch\output\.gitkeep
?????文件????????1536??2019-03-11?21:11??NestedUNet_Pytorch\preprocess_dsb2018.py
?????文件??????????66??2020-03-04?16:26??NestedUNet_Pytorch\README.md
?????文件????????3585??2020-03-04?14:38??NestedUNet_Pytorch\test.py
?????文件????????9591??2020-03-04?14:45??NestedUNet_Pytorch\train.py
?????文件?????????306??2019-03-11?21:11??NestedUNet_Pytorch\utils.py
評論
共有 條評論