Method _computeFeatureMapsRange

tools/quantization/calibration.cpp:809–863

}

void Calibration::_computeFeatureMapsRange() {
    // feed input data according to input images
    int count = 0;

    auto netInfo = _module->getInfo();
    for (const auto& file: _calibrationFiles) {
        std::vector<VARP> inputs;
        for (auto& iter : _featureInfo) {
            iter.second->setVisited(false);
        }
        for (auto& iter : _featureInfo) {
            iter.second->resetUpdatedRangeFlags();
        }

        if (_inputType == Helper::SEQUENCE) {
            mInputs = getModuleInputs(file, netInfo, mInputNames);
        } else {
            auto inputTensor = (MNN::Tensor*)mInputs[0]->getTensor();
            Helper::preprocessInput(_process.get(), _preprocessConfig, file, inputTensor, _inputType, mInputs[0]);
        }

        MNN::TensorCallBackWithInfo before = [&](const std::vector<MNN::Tensor*>& nTensors,
                                             const MNN::OperatorInfo* info) {
            for (auto t : nTensors) {
                if (TensorUtils::getDescribe(t)->index < 0) {
                    continue;
                }
                auto weakPtr = std::weak_ptr<Tensor::InsideDescribe::NativeInsideDescribe>(TensorUtils::getDescribeOrigin(t)->mContent);
                if (_featureInfo.find(weakPtr) != _featureInfo.end()) {
                    if (_featureInfo[weakPtr]->visited() == false) {
                        _featureInfo[weakPtr]->updateRange();
                    }
                }
            }
            return true;
        };
        MNN::TensorCallBackWithInfo after = [&](const std::vector<MNN::Tensor*>& nTensors,
                                                const MNN::OperatorInfo* info) {
            for (auto t : nTensors) {
                if (TensorUtils::getDescribe(t)->index < 0) {
                    continue;
                }
                auto weakPtr = std::weak_ptr<Tensor::InsideDescribe::NativeInsideDescribe>(TensorUtils::getDescribeOrigin(t)->mContent);
                if (_featureInfo.find(weakPtr) != _featureInfo.end()) {
                    if (_featureInfo[weakPtr]->visited() == false) {
                        _featureInfo[weakPtr]->updateRange();
                    }
                }
            }
            return true;
        };
        Express::Executor::getGlobalExecutor()->setCallBack(std::move(before), std::move(after));
        auto outputs = _module->onForward(mInputs);
    }
}

void Calibration::_collectFeatureMapsDistribution() {
    for (auto& iter : _featureInfo) {