wassname/PSPNet-Keras-tensorflow
1
0
Fork 0
mirror of https://github.com/wassname/PSPNet-Keras-tensorflow.git synced 2026-09-12 12:01:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Committed caffe-tensorflow modifications to PSPNet BN layer

Browse Source
This commit is contained in:
VladK
2017-07-13 11:12:02 +03:00
parent d6cfad81d1
commit ed32c508ea
3 changed files with 33 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
caffe-tensorflow/kaffe/layers.py
+3 -1
View File
@@ -11,6 +11,7 @@ LAYER_DESCRIPTORS = {
'Accuracy': shape_scalar,
'ArgMax': shape_not_implemented,
'BatchNorm': shape_identity,
'BN': shape_identity,
'BNLL': shape_not_implemented,
'Concat': shape_concat,
'ContrastiveLoss': shape_scalar,
@@ -18,12 +19,13 @@ LAYER_DESCRIPTORS = {
'Deconvolution': shape_not_implemented,
'Data': shape_data,
'Dropout': shape_identity,
'Interp': shape_identity,
'DummyData': shape_data,
'EuclideanLoss': shape_scalar,
'Eltwise': shape_identity,
'Exp': shape_identity,
'Flatten': shape_not_implemented,
'HDF5Data': shape_data,
'HDF5Datapython': shape_data,
'HDF5Output': shape_identity,
'HingeLoss': shape_scalar,
'Im2col': shape_not_implemented,