wanghaox
7 years ago
parent
67cbb3e3b6
commit
26f03ea13d
@ -1,20 +0,0 @@
|
|||||||
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
|
||||||
|
|
||||||
Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
you may not use this file except in compliance with the License.
|
|
||||||
You may obtain a copy of the License at
|
|
||||||
|
|
||||||
http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing, software
|
|
||||||
distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
See the License for the specific language governing permissions and
|
|
||||||
limitations under the License. */
|
|
||||||
|
|
||||||
#include "paddle/operators/detection_map_op.h"
|
|
||||||
|
|
||||||
namespace ops = paddle::operators;
|
|
||||||
REGISTER_OP_CUDA_KERNEL(
|
|
||||||
detection_map, ops::DetectionMAPOpKernel<paddle::platform::GPUPlace, float>,
|
|
||||||
ops::DetectionMAPOpKernel<paddle::platform::GPUPlace, double>);
|
|
||||||
File diff suppressed because it is too large
Load Diff
@ -1,22 +0,0 @@
|
|||||||
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
|
||||||
|
|
||||||
Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
you may not use this file except in compliance with the License.
|
|
||||||
You may obtain a copy of the License at
|
|
||||||
|
|
||||||
http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing, software
|
|
||||||
distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
See the License for the specific language governing permissions and
|
|
||||||
limitations under the License. */
|
|
||||||
|
|
||||||
#include "paddle/operators/math/detection_util.h"
|
|
||||||
#include "paddle/operators/math/math_function.h"
|
|
||||||
|
|
||||||
namespace paddle {
|
|
||||||
namespace operators {
|
|
||||||
namespace math {} // namespace math
|
|
||||||
} // namespace operators
|
|
||||||
} // namespace paddle
|
|
||||||
@ -1,23 +0,0 @@
|
|||||||
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
|
||||||
|
|
||||||
Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
you may not use this file except in compliance with the License.
|
|
||||||
You may obtain a copy of the License at
|
|
||||||
|
|
||||||
http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing, software
|
|
||||||
distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
See the License for the specific language governing permissions and
|
|
||||||
limitations under the License. */
|
|
||||||
|
|
||||||
#include "paddle/operators/math/detection_util.h"
|
|
||||||
#include "paddle/operators/math/math_function.h"
|
|
||||||
#include "paddle/platform/cuda_helper.h"
|
|
||||||
|
|
||||||
namespace paddle {
|
|
||||||
namespace operators {
|
|
||||||
namespace math {} // namespace math
|
|
||||||
} // namespace operators
|
|
||||||
} // namespace paddle
|
|
||||||
@ -1,128 +0,0 @@
|
|||||||
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.
|
|
||||||
|
|
||||||
Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
you may not use this file except in compliance with the License.
|
|
||||||
You may obtain a copy of the License at
|
|
||||||
|
|
||||||
http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
|
|
||||||
Unless required by applicable law or agreed to in writing, software
|
|
||||||
distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
See the License for the specific language governing permissions and
|
|
||||||
limitations under the License. */
|
|
||||||
#pragma once
|
|
||||||
#include "paddle/framework/selected_rows.h"
|
|
||||||
#include "paddle/platform/device_context.h"
|
|
||||||
|
|
||||||
namespace paddle {
|
|
||||||
namespace operators {
|
|
||||||
namespace math {
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
struct BBox {
|
|
||||||
BBox(T x_min, T y_min, T x_max, T y_max)
|
|
||||||
: x_min(x_min),
|
|
||||||
y_min(y_min),
|
|
||||||
x_max(x_max),
|
|
||||||
y_max(y_max),
|
|
||||||
is_difficult(false) {}
|
|
||||||
|
|
||||||
BBox() {}
|
|
||||||
|
|
||||||
T get_width() const { return x_max - x_min; }
|
|
||||||
|
|
||||||
T get_height() const { return y_max - y_min; }
|
|
||||||
|
|
||||||
T get_center_x() const { return (x_min + x_max) / 2; }
|
|
||||||
|
|
||||||
T get_center_y() const { return (y_min + y_max) / 2; }
|
|
||||||
|
|
||||||
T get_area() const { return get_width() * get_height(); }
|
|
||||||
|
|
||||||
// coordinate of bounding box
|
|
||||||
T x_min;
|
|
||||||
T y_min;
|
|
||||||
T x_max;
|
|
||||||
T y_max;
|
|
||||||
// whether difficult object (e.g. object with heavy occlusion is difficult)
|
|
||||||
bool is_difficult;
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
void GetBBoxFromDetectData(const T* detect_data, const size_t num_bboxes,
|
|
||||||
std::vector<T>& labels, std::vector<T>& scores,
|
|
||||||
std::vector<BBox<T>>& bboxes) {
|
|
||||||
size_t out_offset = bboxes.size();
|
|
||||||
labels.resize(out_offset + num_bboxes);
|
|
||||||
scores.resize(out_offset + num_bboxes);
|
|
||||||
bboxes.resize(out_offset + num_bboxes);
|
|
||||||
for (size_t i = 0; i < num_bboxes; ++i) {
|
|
||||||
labels[out_offset + i] = *(detect_data + i * 7 + 1);
|
|
||||||
scores[out_offset + i] = *(detect_data + i * 7 + 2);
|
|
||||||
BBox<T> bbox;
|
|
||||||
bbox.x_min = *(detect_data + i * 7 + 3);
|
|
||||||
bbox.y_min = *(detect_data + i * 7 + 4);
|
|
||||||
bbox.x_max = *(detect_data + i * 7 + 5);
|
|
||||||
bbox.y_max = *(detect_data + i * 7 + 6);
|
|
||||||
bboxes[out_offset + i] = bbox;
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
void GetBBoxFromLabelData(const T* label_data, const size_t num_bboxes,
|
|
||||||
std::vector<BBox<T>>& bboxes) {
|
|
||||||
size_t out_offset = bboxes.size();
|
|
||||||
bboxes.resize(bboxes.size() + num_bboxes);
|
|
||||||
for (size_t i = 0; i < num_bboxes; ++i) {
|
|
||||||
BBox<T> bbox;
|
|
||||||
bbox.x_min = *(label_data + i * 6 + 1);
|
|
||||||
bbox.y_min = *(label_data + i * 6 + 2);
|
|
||||||
bbox.x_max = *(label_data + i * 6 + 3);
|
|
||||||
bbox.y_max = *(label_data + i * 6 + 4);
|
|
||||||
T is_difficult = *(label_data + i * 6 + 5);
|
|
||||||
if (std::abs(is_difficult - 0.0) < 1e-6)
|
|
||||||
bbox.is_difficult = false;
|
|
||||||
else
|
|
||||||
bbox.is_difficult = true;
|
|
||||||
bboxes[out_offset + i] = bbox;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
inline float JaccardOverlap(const BBox<T>& bbox1, const BBox<T>& bbox2) {
|
|
||||||
if (bbox2.x_min > bbox1.x_max || bbox2.x_max < bbox1.x_min ||
|
|
||||||
bbox2.y_min > bbox1.y_max || bbox2.y_max < bbox1.y_min) {
|
|
||||||
return 0.0;
|
|
||||||
} else {
|
|
||||||
float inter_x_min = std::max(bbox1.x_min, bbox2.x_min);
|
|
||||||
float inter_y_min = std::max(bbox1.y_min, bbox2.y_min);
|
|
||||||
float inter_x_max = std::min(bbox1.x_max, bbox2.x_max);
|
|
||||||
float inter_y_max = std::min(bbox1.y_max, bbox2.y_max);
|
|
||||||
|
|
||||||
float inter_width = inter_x_max - inter_x_min;
|
|
||||||
float inter_height = inter_y_max - inter_y_min;
|
|
||||||
float inter_area = inter_width * inter_height;
|
|
||||||
|
|
||||||
float bbox_area1 = bbox1.get_area();
|
|
||||||
float bbox_area2 = bbox2.get_area();
|
|
||||||
|
|
||||||
return inter_area / (bbox_area1 + bbox_area2 - inter_area);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
bool SortScorePairDescend(const std::pair<float, T>& pair1,
|
|
||||||
const std::pair<float, T>& pair2) {
|
|
||||||
return pair1.first > pair2.first;
|
|
||||||
}
|
|
||||||
|
|
||||||
// template <>
|
|
||||||
// bool SortScorePairDescend(const std::pair<float, NormalizedBBox>& pair1,
|
|
||||||
// const std::pair<float, NormalizedBBox>& pair2) {
|
|
||||||
// return pair1.first > pair2.first;
|
|
||||||
// }
|
|
||||||
|
|
||||||
} // namespace math
|
|
||||||
} // namespace operators
|
|
||||||
} // namespace paddle
|
|
||||||
Loading…
Reference in new issue