Browse Source

fix accuracy of fasterrcnn below 62.5%

pull/15220/head
yuzhenhua 4 years ago
parent
commit
fcd5207416
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      model_zoo/official/cv/faster_rcnn/ascend310_infer/src/AclProcess.cpp
  2. +1
    -1
      model_zoo/official/cv/faster_rcnn/src/FasterRcnn/proposal_generator.py

+ 2
- 2
model_zoo/official/cv/faster_rcnn/ascend310_infer/src/AclProcess.cpp View File

@@ -253,7 +253,7 @@ int AclProcess::Preprocess(const std::string& imageFile) {
resizeOutData.height = modelInfo_.modelHeight;
resizeOutData.width = modelInfo_.modelWidth;
resizeOutData.format = PIXEL_FORMAT_YUV_SEMIPLANAR_420;
ret = dvppCommon_->CombineResizeProcess(decodeOutData, resizeOutData, true, VPC_PT_DEFAULT);
ret = dvppCommon_->CombineResizeProcess(decodeOutData, resizeOutData, true, VPC_PT_PADDING);
if (ret != OK) {
std::cout << "Failed to execute image resized of preprocess module, ret = " << ret << "." << std::endl;
return ret;
@@ -304,7 +304,7 @@ int AclProcess::ModelInfer(std::map<double, double> *costTime_map) {
}

std::vector<void *> inputBuffers({resizeOutData->data, imInfo_dst});
std::vector<size_t> inputSizes({resizeOutData->dataSize, 4 * 2});
std::vector<size_t> inputSizes({resizeOutData->dataSize, 4 * 4});

for (size_t i = 0; i < modelInfo_.outputNum; i++) {
aclrtMemset(outputBuffers_[i], outputSizes_[i], 0, outputSizes_[i]);


+ 1
- 1
model_zoo/official/cv/faster_rcnn/src/FasterRcnn/proposal_generator.py View File

@@ -131,7 +131,7 @@ class Proposal(nn.Cell):

self.topKv2 = P.TopK(sorted=True)
self.topK_shape_stage2 = (self.max_num, 1)
self.min_float_num = -65536.0
self.min_float_num = -65500.0
self.topK_mask = Tensor(self.min_float_num * np.ones(total_max_topk_input, np.float32))

def construct(self, rpn_cls_score_total, rpn_bbox_pred_total, anchor_list):


Loading…
Cancel
Save