fix strided_slice layer: error of param end

minglu2019 · minglu2019 · commit c877b70a48ec · 2020-07-20T11:51:31.000+02:00
diff --git a/src/caffe/layers/strided_slice_layer.cpp b/src/caffe/layers/strided_slice_layer.cpp
@@ -124,7 +124,7 @@ void StridedSliceLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
       if (strided_end_[i] >= 0) {
         strided_end_[i] = std::min(strided_end_[i], b_shape[i]);
       } else {
-        strided_end_[i] = std::max(strided_end_[i] + b_shape[i], 0);
+        strided_end_[i] = std::max(strided_end_[i] + b_shape[i], -1);
       }
     }
     // begin_mask and end_mask
@@ -188,7 +188,7 @@ void StridedSliceLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
       if (strided_end_[i] >= 0) {
         strided_end_[i] = std::min(strided_end_[i], b_shape[axes_[i]]);
       } else {
-        strided_end_[i] = std::max(strided_end_[i] + b_shape[axes_[i]], 0);
+        strided_end_[i] = std::max(strided_end_[i] + b_shape[axes_[i]], -1);
       }
       // caculate top shape
       if (strides_[i] > 0) {

Original file line number	Diff line number	Diff line change
`@@ -124,7 +124,7 @@ void StridedSliceLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,`
`124`	`124`	`if (strided_end_[i] >= 0) {`
`125`	`125`	`strided_end_[i] = std::min(strided_end_[i], b_shape[i]);`
`126`	`126`	`} else {`
`127`		`- strided_end_[i] = std::max(strided_end_[i] + b_shape[i], 0);`
	`127`	`+ strided_end_[i] = std::max(strided_end_[i] + b_shape[i], -1);`
`128`	`128`	`}`
`129`	`129`	`}`
`130`	`130`	`// begin_mask and end_mask`
`@@ -188,7 +188,7 @@ void StridedSliceLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,`
`188`	`188`	`if (strided_end_[i] >= 0) {`
`189`	`189`	`strided_end_[i] = std::min(strided_end_[i], b_shape[axes_[i]]);`
`190`	`190`	`} else {`
`191`		`- strided_end_[i] = std::max(strided_end_[i] + b_shape[axes_[i]], 0);`
	`191`	`+ strided_end_[i] = std::max(strided_end_[i] + b_shape[axes_[i]], -1);`
`192`	`192`	`}`
`193`	`193`	`// caculate top shape`
`194`	`194`	`if (strides_[i] > 0) {`