mirror of
https://github.com/Karllzy/cotton_color.git
synced 2025-11-09 03:03:53 +00:00
添加拖延
This commit is contained in:
parent
f9609dbd34
commit
3f21a63291
@ -7,70 +7,114 @@
|
||||
|
||||
#include "utils.h"
|
||||
|
||||
std::vector<std::vector<uint8_t>> generateMaskFromImage(const MIL_ID& inputImage, int widthBlocks, int heightBlocks, int threshold = 10, int rowRange = 50) {
|
||||
// 读取图像
|
||||
cv::Mat image = mil2mat(inputImage);
|
||||
//
|
||||
// // 检查图像是否成功读取
|
||||
// if (image.empty()) {
|
||||
// std::cerr << "无法加载图像，请检查路径是否正确: " << imagePath << std::endl;
|
||||
// exit(EXIT_FAILURE);
|
||||
// }
|
||||
#include <opencv2/opencv.hpp>
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
using namespace std;
|
||||
|
||||
// 确保图像是二值化的
|
||||
// Function to convert an image into a mask based on sizeThreshold
|
||||
vector<vector<uint8_t>> generateMask(
|
||||
const MIL_ID& inputImg,
|
||||
int outputWidth,
|
||||
int outputHeight,
|
||||
int sizeThreshold
|
||||
) {
|
||||
cv::Mat image = mil2mat(inputImg);
|
||||
|
||||
// Ensure the image is binary
|
||||
cv::threshold(image, image, 128, 255, cv::THRESH_BINARY);
|
||||
|
||||
// 获取图像的宽度和高度
|
||||
int imageWidth = image.cols;
|
||||
int imageHeight = image.rows;
|
||||
|
||||
// 计算每个块的宽度和高度
|
||||
int blockWidth = imageWidth / widthBlocks;
|
||||
int blockHeight = imageHeight / heightBlocks;
|
||||
int blockWidth = imageWidth / outputWidth;
|
||||
int blockHeight = imageHeight / outputHeight;
|
||||
|
||||
// 创建掩膜矩阵 (uint8_t 类型)
|
||||
std::vector<std::vector<uint8_t>> mask(heightBlocks, std::vector<uint8_t>(widthBlocks, 0));
|
||||
vector<vector<uint8_t>> mask(outputHeight, vector<uint8_t>(outputWidth, 0));
|
||||
|
||||
// 遍历每个块并统计白色像素点的数量
|
||||
for (int i = 0; i < heightBlocks; ++i) {
|
||||
for (int j = 0; j < widthBlocks; ++j) {
|
||||
// 计算块的起始和结束位置
|
||||
for (int i = 0; i < outputHeight; ++i) {
|
||||
for (int j = 0; j < outputWidth; ++j) {
|
||||
int x_start = j * blockWidth;
|
||||
int y_start = i * blockHeight;
|
||||
int x_end = (j == widthBlocks - 1) ? imageWidth : (j + 1) * blockWidth;
|
||||
int y_end = (i == heightBlocks - 1) ? imageHeight : (i + 1) * blockHeight;
|
||||
int x_end = (j == outputWidth - 1) ? imageWidth : (j + 1) * blockWidth;
|
||||
int y_end = (i == outputHeight - 1) ? imageHeight : (i + 1) * blockHeight;
|
||||
|
||||
// 提取当前块
|
||||
cv::Mat block = image(cv::Rect(x_start, y_start, x_end - x_start, y_end - y_start));
|
||||
|
||||
// 统计块中白色像素的数量
|
||||
int whitePixelCount = cv::countNonZero(block);
|
||||
|
||||
// 如果白色像素数大于阈值，将该块标记为 255
|
||||
if (whitePixelCount > threshold) {
|
||||
if (whitePixelCount > sizeThreshold) {
|
||||
mask[i][j] = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 遍历每一列，处理规则：当某列出现第一个1时，将其后rowRange行全部置为255
|
||||
for (int j = 0; j < widthBlocks; ++j) {
|
||||
int i = 0;
|
||||
while (i < heightBlocks) {
|
||||
if (mask[i][j] == 1) {
|
||||
// 找到第一个1，处理后面rowRange行
|
||||
for (int k = i; k < std::min(i + rowRange, heightBlocks); ++k) {
|
||||
mask[k][j] = 1;
|
||||
}
|
||||
// 跳过已经设置为255的后rowRange行
|
||||
i += rowRange;
|
||||
} else {
|
||||
// 如果当前位置为0，则继续检测下一行
|
||||
++i;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return mask;
|
||||
}
|
||||
pair<vector<vector<uint8_t>>, vector<vector<uint8_t>>> applyRowRangeDelay(
|
||||
const vector<vector<uint8_t>>& mask,
|
||||
const vector<vector<uint8_t>>& tail,
|
||||
int rowRange
|
||||
) {
|
||||
int outputHeight = (int)mask.size();
|
||||
int outputWidth = (int)mask[0].size();
|
||||
|
||||
vector<vector<uint8_t>> mask_after_row_range(outputHeight, vector<uint8_t>(outputWidth, 0));
|
||||
vector<vector<uint8_t>> newTail(rowRange, vector<uint8_t>(outputWidth, 0));
|
||||
|
||||
// 先将旧的 tail 映射到 mask_after_row_range 的顶部几行
|
||||
for (int i = 0; i < (int)tail.size(); ++i) {
|
||||
for (int j = 0; j < outputWidth; ++j) {
|
||||
if (i < outputHeight) {
|
||||
mask_after_row_range[i][j] = max(mask_after_row_range[i][j], tail[i][j]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 对当前 mask 应用 rowRange 的拖影效果
|
||||
for (int j = 0; j < outputWidth; ++j) {
|
||||
for (int i = 0; i < outputHeight; ++i) {
|
||||
if (mask[i][j] == 1) {
|
||||
// 从当前行 i 开始，向下扩展 rowRange 行
|
||||
int end_line = i + rowRange - 1;
|
||||
|
||||
// 先处理仍在 mask 范围内的部分
|
||||
int inside_mask_end = min(end_line, outputHeight - 1);
|
||||
for (int line = i; line <= inside_mask_end; ++line) {
|
||||
mask_after_row_range[line][j] = 1;
|
||||
}
|
||||
|
||||
// 超出 mask 范围的行进入 tail
|
||||
if (end_line >= outputHeight) {
|
||||
// 从 outputHeight 行开始的部分属于 tail
|
||||
for (int line = outputHeight; line <= end_line; ++line) {
|
||||
int tail_line_idx = line - outputHeight;
|
||||
if (tail_line_idx >= 0 && tail_line_idx < (int)newTail.size()) {
|
||||
newTail[tail_line_idx][j] = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return {mask_after_row_range, newTail};
|
||||
}
|
||||
|
||||
|
||||
// Updated wrapper function
|
||||
pair<vector<vector<uint8_t>>, vector<vector<uint8_t>>> generateMaskWithTail(
|
||||
const MIL_ID& inputImg,
|
||||
const vector<vector<uint8_t>>& tail,
|
||||
int outputWidth,
|
||||
int outputHeight,
|
||||
int sizeThreshold = 10,
|
||||
int rowRange = 50
|
||||
) {
|
||||
// Generate the mask from the image
|
||||
vector<vector<uint8_t>> mask = generateMask(inputImg, outputWidth, outputHeight, sizeThreshold);
|
||||
|
||||
// Apply rowRange delay
|
||||
return applyRowRangeDelay(mask, tail, rowRange);
|
||||
}
|
||||
|
||||
|
||||
@ -10,7 +10,16 @@
|
||||
#include <vector>
|
||||
#include"mil.h"
|
||||
#include <iostream>
|
||||
using namespace std;
|
||||
|
||||
|
||||
std::vector<std::vector<uint8_t>> generateMaskFromImage(const MIL_ID& inputImage, int widthBlocks, int heightBlocks, int threshold , int rowRange) ;
|
||||
pair<vector<vector<uint8_t>>, vector<vector<uint8_t>>> generateMaskWithTail(
|
||||
const MIL_ID& inputImg,
|
||||
const vector<vector<uint8_t>>& tail,
|
||||
int outputWidth,
|
||||
int outputHeight,
|
||||
int sizeThreshold,
|
||||
int rowRange
|
||||
);
|
||||
|
||||
#endif //MASK_H
|
||||
|
||||
@ -18,7 +18,7 @@ target_link_libraries(test_template_matching Matrox ${OpenCV_LIBS} ${MIL_LIBS})
|
||||
add_executable(test_mask
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/test_mask.cpp)
|
||||
# 链接 Matrox 模块和依赖库
|
||||
target_link_libraries(test_mask Matrox ${OpenCV_LIBS} ${MIL_LIBS})
|
||||
target_link_libraries(test_mask Matrox CVDL ${OpenCV_LIBS} ${MIL_LIBS})
|
||||
|
||||
#测试用例 4：test_onnx
|
||||
add_executable(test_onnx
|
||||
|
||||
@ -1,49 +1,188 @@
|
||||
//
|
||||
// Created by zjc on 24-11-26.
|
||||
//
|
||||
// #include <Matrox/utils.h>
|
||||
|
||||
#include "vector"
|
||||
#include"iostream"
|
||||
#include"string"
|
||||
#include"Matrox/mask.h"
|
||||
#include "iostream"
|
||||
#include "string"
|
||||
#include "Matrox/mask.h"
|
||||
#include "CVDL/OnnxRunner.h"
|
||||
#include "mil.h"
|
||||
|
||||
#include"mil.h"
|
||||
#define Image_PATH3 MIL_TEXT("C:\\Users\\zjc\\Desktop\\mask_5.png")
|
||||
#define Image_PATH0 MIL_TEXT("test_imgs\\test_mask_imgs\\mask_1.png")
|
||||
#define Image_PATH1 MIL_TEXT("test_imgs\\test_mask_imgs\\mask_6.png")
|
||||
MIL_ID MilApplication, MilSystem, MilDisplay;
|
||||
|
||||
int main() {
|
||||
// 指定图像路径
|
||||
// std::string imagePath = "C:\\Users\\zjc\\Desktop\\mask_1.png";
|
||||
MappAllocDefault(M_DEFAULT, &MilApplication, &MilSystem, &MilDisplay, M_NULL, M_NULL);
|
||||
MIL_ID MilImage = M_NULL, MilHighSatResult = M_NULL, MilTemplateMatchingResult = M_NULL;
|
||||
MbufRestore(Image_PATH3, MilSystem, &MilImage);
|
||||
// std::unordered_map<std::string, int> config = loadConfig("C:\\Users\\zjc\\Desktop\\config\\mask_config.txt");
|
||||
// // 设置分块数量和白色像素点阈值
|
||||
// int widthBlocks = config["widthBlocks"];
|
||||
// int heightBlocks = config["heightBlocks"];
|
||||
// int threshold = config["threshold"];
|
||||
|
||||
// 输出参数值
|
||||
// std::cout << "widthBlocks: " << widthBlocks << std::endl;
|
||||
|
||||
int widthBlocks = 24;
|
||||
int heightBlocks = 1024;
|
||||
int heightBlocks = 96;
|
||||
int threshold = 20;
|
||||
int rowRange = 50; // 后续50行设置为
|
||||
int rowRange = 1;
|
||||
|
||||
// 生成掩膜
|
||||
std::vector<std::vector<uint8_t>> mask = generateMaskFromImage(MilImage, widthBlocks, heightBlocks, threshold, rowRange);
|
||||
std::vector<std::vector<uint8_t>> tail(0);
|
||||
|
||||
// 打印掩膜结果
|
||||
for (int i = 0; i < heightBlocks; ++i) {
|
||||
for (int j = 0; j < widthBlocks; ++j) {
|
||||
std::cout << (int)mask[i][j] << " ";
|
||||
std::cout << "=============================row 632, 1020 = 1 ===========================================" << std::endl;
|
||||
for (int imgIndex = 0; imgIndex < 2; imgIndex++) {
|
||||
MIL_ID MilImage = M_NULL;
|
||||
if (imgIndex == 0)
|
||||
MbufRestore(Image_PATH0, MilSystem, &MilImage);
|
||||
else
|
||||
MbufRestore(Image_PATH1, MilSystem, &MilImage);
|
||||
|
||||
Timer timer1;
|
||||
timer1.restart();
|
||||
auto [mask, newTail] = generateMaskWithTail(MilImage, tail, widthBlocks, heightBlocks, threshold, rowRange);
|
||||
tail = newTail;
|
||||
timer1.printElapsedTime("Convert to mask spent");
|
||||
|
||||
// 打印mask结果
|
||||
for (int r = 0; r < heightBlocks; ++r) {
|
||||
std::cout << "mask Row: " << r << " ";
|
||||
for (int c = 0; c < widthBlocks; ++c) {
|
||||
std::cout << (int)mask[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
MbufFree(MilImage);
|
||||
MappFreeDefault(MilApplication, MilSystem, MilDisplay, M_NULL, M_NULL);
|
||||
if(imgIndex==0)
|
||||
{
|
||||
|
||||
|
||||
// 打印tail结果（使用tail.size()进行循环，避免越界）
|
||||
for (size_t r = 0; r < tail.size(); ++r) {
|
||||
std::cout << "tail 0 Row: " << r << " ";
|
||||
for (size_t c = 0; c < tail[r].size(); ++c) {
|
||||
std::cout << (int)tail[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
}
|
||||
MbufFree(MilImage);
|
||||
}
|
||||
|
||||
std::cout << "=============================row 1020 = 1 ===========================================" << std::endl;
|
||||
rowRange = 10;
|
||||
tail = std::vector<std::vector<uint8_t>>(rowRange, std::vector<uint8_t>(widthBlocks, 0));
|
||||
for (int imgIndex = 0; imgIndex < 2; imgIndex++) {
|
||||
MIL_ID MilImage = M_NULL;
|
||||
if (imgIndex == 0)
|
||||
MbufRestore(Image_PATH0, MilSystem, &MilImage);
|
||||
else
|
||||
MbufRestore(Image_PATH1, MilSystem, &MilImage);
|
||||
|
||||
Timer timer1;
|
||||
timer1.restart();
|
||||
auto [mask, newTail] = generateMaskWithTail(MilImage, tail, widthBlocks, heightBlocks, threshold, rowRange);
|
||||
tail = newTail;
|
||||
timer1.printElapsedTime("Convert to mask spent");
|
||||
|
||||
// 打印mask结果
|
||||
for (int r = 0; r < heightBlocks; ++r) {
|
||||
std::cout << "Row: " << r << " ";
|
||||
for (int c = 0; c < widthBlocks; ++c) {
|
||||
std::cout << (int)mask[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
|
||||
if(imgIndex==0)
|
||||
{
|
||||
// 打印tail结果（使用tail.size()进行循环，避免越界）
|
||||
for (size_t r = 0; r < tail.size(); ++r) {
|
||||
std::cout << "tail 0 Row: " << r << " ";
|
||||
for (size_t c = 0; c < tail[r].size(); ++c) {
|
||||
std::cout << (int)tail[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
MbufFree(MilImage);
|
||||
}
|
||||
|
||||
std::cout << "=============================row 1030 = 1 ===========================================" << std::endl;
|
||||
rowRange = 20;
|
||||
tail = std::vector<std::vector<uint8_t>>(rowRange, std::vector<uint8_t>(widthBlocks, 0));
|
||||
for (int imgIndex = 0; imgIndex < 2; imgIndex++) {
|
||||
MIL_ID MilImage = M_NULL;
|
||||
if (imgIndex == 0)
|
||||
MbufRestore(Image_PATH0, MilSystem, &MilImage);
|
||||
else
|
||||
MbufRestore(Image_PATH1, MilSystem, &MilImage);
|
||||
|
||||
Timer timer1;
|
||||
timer1.restart();
|
||||
auto [mask, newTail] = generateMaskWithTail(MilImage, tail, widthBlocks, heightBlocks, threshold, rowRange);
|
||||
tail = newTail;
|
||||
timer1.printElapsedTime("Convert to mask spent");
|
||||
|
||||
// 打印mask结果
|
||||
for (int r = 0; r < heightBlocks; ++r) {
|
||||
std::cout << "Row: " << r << " ";
|
||||
for (int c = 0; c < widthBlocks; ++c) {
|
||||
std::cout << (int)mask[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
|
||||
// 打印tail结果
|
||||
if(imgIndex==0)
|
||||
{
|
||||
// 打印tail结果（使用tail.size()进行循环，避免越界）
|
||||
for (size_t r = 0; r < tail.size(); ++r) {
|
||||
std::cout << "tail 0 Row: " << r << " ";
|
||||
for (size_t c = 0; c < tail[r].size(); ++c) {
|
||||
std::cout << (int)tail[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
MbufFree(MilImage);
|
||||
}
|
||||
|
||||
std::cout << "=============================row 1030 = 1 ===========================================" << std::endl;
|
||||
rowRange = 40;
|
||||
tail = std::vector<std::vector<uint8_t>>(rowRange, std::vector<uint8_t>(widthBlocks, 0));
|
||||
for (int imgIndex = 0; imgIndex < 2; imgIndex++) {
|
||||
MIL_ID MilImage = M_NULL;
|
||||
if (imgIndex == 0)
|
||||
MbufRestore(Image_PATH0, MilSystem, &MilImage);
|
||||
else
|
||||
MbufRestore(Image_PATH1, MilSystem, &MilImage);
|
||||
|
||||
Timer timer1;
|
||||
timer1.restart();
|
||||
auto [mask, newTail] = generateMaskWithTail(MilImage, tail, widthBlocks, heightBlocks, threshold, rowRange);
|
||||
tail = newTail;
|
||||
timer1.printElapsedTime("Convert to mask spent");
|
||||
|
||||
// 打印mask结果
|
||||
for (int r = 0; r < heightBlocks; ++r) {
|
||||
std::cout << "Row: " << r << " ";
|
||||
for (int c = 0; c < widthBlocks; ++c) {
|
||||
std::cout << (int)mask[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
|
||||
// 打印tail结果
|
||||
if(imgIndex==0)
|
||||
{
|
||||
// 打印tail结果（使用tail.size()进行循环，避免越界）
|
||||
for (size_t r = 0; r < tail.size(); ++r) {
|
||||
std::cout << "tail 0 Row: " << r << " ";
|
||||
for (size_t c = 0; c < tail[r].size(); ++c) {
|
||||
std::cout << (int)tail[r][c] << " ";
|
||||
}
|
||||
std::cout << std::endl;
|
||||
}
|
||||
}
|
||||
|
||||
MbufFree(MilImage);
|
||||
}
|
||||
|
||||
MappFreeDefault(MilApplication, MilSystem, MilDisplay, M_NULL, M_NULL);
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user