Method preprocessInput

tools/quantization/Helper.cpp:140–224

}

void Helper::preprocessInput(MNN::CV::ImageProcess* pretreat, PreprocessConfig preprocessConfig, const std::string& filename, MNN::Tensor* input, InputType inputType, MNN::Express::VARP var) {
    if (inputType == InputType::IMAGE) {
        int originalWidth, originalHeight, comp;
        auto bitmap32bits = stbi_load(filename.c_str(), &originalWidth, &originalHeight, &comp, 4);

        DCHECK(bitmap32bits != nullptr) << "input image error!";

        const int hCropSize = int(originalHeight * preprocessConfig.centerCropHeight);
        const int wCropSize = int(originalWidth * preprocessConfig.centerCropWidth);
        MNN_ASSERT(hCropSize > 0 && wCropSize > 0);
        // default center crop
        int startH = (originalHeight - hCropSize) / 2;
        int startW = (originalWidth - wCropSize) / 2;

        const int endH = startH + hCropSize;
        const int endW = startW + wCropSize;

        float srcPoints[] = {
                float(startW), float(startH),
                float(startW), float(endH - 1),
                float(endW - 1), float(startH),
                float(endW - 1), float(endH - 1),
        };
        const int oh = preprocessConfig.targetHeight;
        const int ow = preprocessConfig.targetWidth;

        float dstPoints[] = {
                0.0f, 0.0f,
                0.0f, float(oh - 1),
                float(ow - 1), 0.0f,
                float(ow - 1), float(oh - 1),
        };
        MNN::CV::Matrix trans;
        trans.setPolyToPoly((MNN::CV::Point*)dstPoints, (MNN::CV::Point*)srcPoints, 4);

        pretreat->setMatrix(trans);
        pretreat->convert(bitmap32bits, originalWidth, originalHeight, 0, input);

        stbi_image_free(bitmap32bits);
    }
    if (inputType == InputType::SEQUENCE) {
        if (!stringEndWith(filename, ".txt")) {
            MNN_ERROR("Error: only '.txt' files are supported for sequence input.\n");
            return;
        }

        std::ifstream f(filename);
        if (!f.is_open()) {
            MNN_ERROR("open file %s failed.\n", filename.c_str());
            return;
        }

        std::string line;
        std::vector<std::vector<float> > rawData;
        while (std::getline(f, line)) {
            std::stringstream ss(line);
            float v;
            std::vector<float> lineData;