fix one_hot layer

minglu2019 · minglu2019 · commit 421eace9222f · 2020-05-27T10:36:35.000+02:00
diff --git a/src/caffe/layers/one_hot_layer.cpp b/src/caffe/layers/one_hot_layer.cpp
@@ -85,7 +85,7 @@ void OneHotLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype> *> &bottom,
       vector<int> b_ind = indices(i, bottom_shape);
       int b_idx = offset(top, axis_ind, b_ind);
       if (bottom_data[i] >= 0 && bottom_data[i] < depth) {
-        int t_idx = b_idx + bottom_data[i] * top[0]->count(axis + 1);
+        int t_idx = b_idx + int(bottom_data[i]) * top[0]->count(axis + 1);
         top_data[t_idx] = on_value;
       }
     }

Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,7 @@ void OneHotLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype> *> &bottom,`
`85`	`85`	`vector<int> b_ind = indices(i, bottom_shape);`
`86`	`86`	`int b_idx = offset(top, axis_ind, b_ind);`
`87`	`87`	`if (bottom_data[i] >= 0 && bottom_data[i] < depth) {`
`88`		`- int t_idx = b_idx + bottom_data[i] * top[0]->count(axis + 1);`
	`88`	`+ int t_idx = b_idx + int(bottom_data[i]) * top[0]->count(axis + 1);`
`89`	`89`	`top_data[t_idx] = on_value;`
`90`	`90`	`}`
`91`	`91`	`}`