nihui
/
ncnn
Not watched
1
0
Fork 0
Code
Releases 46 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

fix layer pooling when elemsize is 1 (#1705)

tags/20200616
Xu Yang GitHub 6 years ago
parent
e49a9ec736
commit
3a34b99fe6
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      src/layer/pooling.cpp

+ 1
- 1
src/layer/pooling.cpp View File

@@ -272,7 +272,7 @@ void Pooling::make_padding(const Mat& bottom_blob, Mat& bottom_blob_bordered, co
float pad_value = 0.f;
if (pooling_type == PoolMethod_MAX)
{
pad_value = -FLT_MAX;
pad_value = bottom_blob.elemsize == 1 ? -128.f : -FLT_MAX;
}
else if (pooling_type == PoolMethod_AVE)
{


Write Preview
Loading…
Cancel
Save