dearsky/Deep_Learning
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
shadow
2020-05-02 17:28:40 +08:00
parent 0b38565bc1
commit e197adbf29
4 changed files with 68 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train/train.py
View File

@@ -33,7 +33,7 @@ def bn_relu(x):
return x
def resner50(out_dims, input_shape=(128, 128, 1)):
def resnet50(out_dims, input_shape=(128, 128, 1)):
# input_dim = Input(input_shape)
resnet_base_model = ResNet50(include_top=False, weights=None, input_shape=input_shape)