zhangwei
/
VTrain
generated from zhangwei/Matrixai
1
0
Fork 1
Code Issues Pull Requests Packages Projects 1 Releases 1 Wiki Activity

基础版

This commit is contained in:
Mr.V 2024-11-27 21:47:42 +08:00
parent b5b3e34735
commit c31a201a78
3 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -1,2 +1,3 @@
# Train_Identify # Train_Identify
支持单摄像头识别，支持集装箱号识别

10
config/matrix.yaml
View File

@ -32,21 +32,11 @@ connects:
MoveEngine_0_1: "SaveMoveInfoEngine_0_0 1024" MoveEngine_0_1: "SaveMoveInfoEngine_0_0 1024"
SaveMoveImageEngine_0_0: "TrainStep1DataReadEngine_0_0 1024" SaveMoveImageEngine_0_0: "TrainStep1DataReadEngine_0_0 1024"
TrainStep1DataReadEngine_0_0: "TrainStep1InferenceEngine_0_0 1024" TrainStep1DataReadEngine_0_0: "TrainStep1InferenceEngine_0_0 1024"
# TrainStep1DataReadEngine_0_1: "ContainerStep1InferenceEngine_0_0 1024"
# TrainStep1InferenceEngine_0_0: "Step1MergeEngine_0_0 1024"
# ContainerStep1InferenceEngine_0_0: "Step1MergeEngine_0_1 1024"
# Step1MergeEngine_0_0: "TrainStep1FilterEngine_0_0 1024"
TrainStep1InferenceEngine_0_0: "TrainStep1FilterEngine_0_0 1024" TrainStep1InferenceEngine_0_0: "TrainStep1FilterEngine_0_0 1024"
TrainStep1FilterEngine_0_0: "TrainDivideEngine_0_0 1024" TrainStep1FilterEngine_0_0: "TrainDivideEngine_0_0 1024"
TrainDivideEngine_0_0: "TrainStep2DataReadEngine_0_0 1024" TrainDivideEngine_0_0: "TrainStep2DataReadEngine_0_0 1024"
TrainStep2DataReadEngine_0_0: "TrainStep2InferenceEngine_0_0 1024" TrainStep2DataReadEngine_0_0: "TrainStep2InferenceEngine_0_0 1024"
# TrainStep2DataReadEngine_0_1: "ContainerStep2InferenceEngine_0_0 1024"
TrainStep2InferenceEngine_0_0: "TrainCharacterConversionEngine_0_0 1024" TrainStep2InferenceEngine_0_0: "TrainCharacterConversionEngine_0_0 1024"
# ContainerStep2InferenceEngine_0_0: "ContainerCharacterConversionEngine_0_0 1024"
# TrainCharacterConversionEngine_0_0: "Step2MergeEngine_0_0 1024"
# ContainerCharacterConversionEngine_0_0: "Step2MergeEngine_0_1 1024"
# Step2MergeEngine_0_0: "SelectBestEngine_0_0 1024"
# Step2MergeEngine_0_1: "SaveDebugImageEngine_0_0 1024"
TrainCharacterConversionEngine_0_0: "SelectBestEngine_0_0 1024" TrainCharacterConversionEngine_0_0: "SelectBestEngine_0_0 1024"
TrainCharacterConversionEngine_0_1: "SaveDebugImageEngine_0_0 1024" TrainCharacterConversionEngine_0_1: "SaveDebugImageEngine_0_0 1024"
SelectBestEngine_0_0: "SaveResultCSVEngine_0_0 1024" SelectBestEngine_0_0: "SaveResultCSVEngine_0_0 1024"

2
engine/Step2DataReadEngine/TrainStep2DataReadEngine.cpp
View File

@ -165,7 +165,7 @@ APP_ERROR TrainStep2DataReadEngine::Process()
pVStep2InputData->vecSingleData = vecSingleData; pVStep2InputData->vecSingleData = vecSingleData;
pVStep2InputData->iFrameId = iFrameId; pVStep2InputData->iFrameId = iFrameId;
pVStep2InputData->cvImage = cv::imread(strImagePath); pVStep2InputData->cvImage = cv::imread(strImagePath);
pVStep2InputData->bIsEnd = jvFrameInfo.isMember("isEnd") ? jvFrameInfo["isEnd"].asBool() : true; pVStep2InputData->bIsEnd = !jvFrameInfo.isMember("isEnd") || jvFrameInfo["isEnd"].asBool();
outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(pVStep2InputData), true); outputQueMap_[strPort0_]->push(std::static_pointer_cast<void>(pVStep2InputData), true);
// outputQueMap_[strPort1_]->push(std::static_pointer_cast<void>(pVStep2InputData), true); // outputQueMap_[strPort1_]->push(std::static_pointer_cast<void>(pVStep2InputData), true);