diff --git a/mace/kernels/proposal.h b/mace/kernels/proposal.h index 897e1915704f7681615773bf970e2aac595ec9bd..d3afe966e2da1892f7fc3862fb1e49cbe14ea082 100644 --- a/mace/kernels/proposal.h +++ b/mace/kernels/proposal.h @@ -195,7 +195,6 @@ struct ProposalFunctor { proposals[sanc_idx][3] = std::max( std::min(pred_ctr_y + pred_h / 2, im_height), 0); - } } } diff --git a/mace/kernels/psroi_align.h b/mace/kernels/psroi_align.h index 9d6f43118ad3dd6cc947a2d92d73518abe97ac7f..57e19c3c0a8fd297a46f9b47d2a2ff83e7dd0afe 100644 --- a/mace/kernels/psroi_align.h +++ b/mace/kernels/psroi_align.h @@ -5,6 +5,9 @@ #ifndef MACE_KERNELS_PSROI_ALIGN_H_ #define MACE_KERNELS_PSROI_ALIGN_H_ +#include +#include + #include "mace/core/future.h" #include "mace/core/tensor.h" #include "mace/public/mace.h" @@ -169,7 +172,7 @@ struct PSROIAlignFunctor { const int group_size_; }; -} // namepsace kernels -} // namespace mace +} // namespace kernels +} // namespace mace -#endif // MACE_KERNELS_PSROI_ALIGN_H_ +#endif // MACE_KERNELS_PSROI_ALIGN_H_ diff --git a/mace/ops/proposal_test.cc b/mace/ops/proposal_test.cc index 11eb7a49b94a5543c5f70ebea954763dcbbabac4..94203deb4b82bc1a1d2d96e3a122e6856f0ce109 100644 --- a/mace/ops/proposal_test.cc +++ b/mace/ops/proposal_test.cc @@ -4,7 +4,6 @@ #include "mace/core/operator.h" #include "mace/ops/ops_test_util.h" -#include namespace mace { namespace ops { @@ -54,7 +53,6 @@ TEST_F(ProposalOpTest, CPUSimple) { auto expected_tensor = CreateTensor({1, 1, 1, 5}, {0, 0, 0, 255, 255}); ExpectTensorNear(*expected_tensor, *net.GetTensor("Output"), 1e-5); - } diff --git a/mace/ops/psroi_align.h b/mace/ops/psroi_align.h index c81674205822d16d793cf94f98bea6556c227d6f..e134b4a43cb78022895ee62a7b3ccfde3c5e110d 100644 --- a/mace/ops/psroi_align.h +++ b/mace/ops/psroi_align.h @@ -38,7 +38,7 @@ class PSROIAlignOp : public Operator { OP_OUTPUT_TAGS(OUTPUT); }; -} // namespace ops -} // namespace mace +} // namespace ops +} // namespace mace -#endif // MACE_OPS_PSROI_ALIGN_H_ +#endif // MACE_OPS_PSROI_ALIGN_H_