Doni/safesight-edge
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow graph executor tuning from config

Browse Source
This commit is contained in:
tian 2026-03-15 01:59:26 +08:00
parent 53afc006c3
commit a32f86fca1
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
configs/sample_person_shoe_two_stage_balanced.json
View File

@ -6,6 +6,10 @@
"graphs": [ "graphs": [
{ {
"name": "person_shoe_two_stage_balanced", "name": "person_shoe_two_stage_balanced",
"executor": {
"batch_size": 2,
"run_budget": 8
},
"nodes": [ "nodes": [
{ {
"id": "in", "id": "in",
@ -139,6 +143,7 @@
"type": "publish", "type": "publish",
"role": "filter", "role": "filter",
"enable": true, "enable": true,
"cpu_affinity": [7],
"codec": "h264", "codec": "h264",
"fps": 30, "fps": 30,
"bitrate_kbps": 2000, "bitrate_kbps": 2000,

9
src/graph_manager.cpp
View File

@ -199,7 +199,14 @@ bool EffectiveQueueForEdge(const ParsedEdge& e, const SimpleJson* from_node_queu
class Graph::Executor { class Graph::Executor {
public: public:
explicit Executor(Graph& g) explicit Executor(Graph& g)
: g_(g), batch_size_(DefaultGraphBatchSize()), run_budget_(DefaultGraphRunBudget()) {} : g_(g), batch_size_(DefaultGraphBatchSize()), run_budget_(DefaultGraphRunBudget()) {
if (const SimpleJson* exec = g_.graph_cfg_.Find("executor"); exec && exec->IsObject()) {
const int batch = exec->ValueOr<int>("batch_size", static_cast<int>(batch_size_));
const int budget = exec->ValueOr<int>("run_budget", static_cast<int>(run_budget_));
if (batch > 0 && batch <= 1024) batch_size_ = static_cast<size_t>(batch);
if (budget > 0 && budget <= 4096) run_budget_ = static_cast<size_t>(budget);
}
}
bool Start() { bool Start() {
Stop(); Stop();