remove the YOLO-World open-vocabulary engine
Deletes the yoloworld_onnx engine, the CLIP BPE tokenizer, the standalone test binary and both export scripts. Drops the update_vocabulary input and the yoloworld_onnx/open_vocab dispatch from main.cc, along with the now unused Engine::set_labels. Documentation follows.
This commit is contained in:
-18
@@ -1,6 +1,5 @@
|
||||
#include "engines/engine.h"
|
||||
#include "engines/yolo_onnx.h"
|
||||
#include "engines/yoloworld_onnx.h"
|
||||
#include "engines/zed_native.h"
|
||||
#include "util/env.h"
|
||||
#include "zed/camera.h"
|
||||
@@ -78,9 +77,6 @@ int main() {
|
||||
if (engine_name == "yolo_onnx" || engine_name == "onnx") {
|
||||
auto cfg = engines::YoloOnnxConfig::from_env();
|
||||
detector = std::make_unique<engines::YoloOnnx>(cfg);
|
||||
} else if (engine_name == "yoloworld_onnx" || engine_name == "open_vocab") {
|
||||
auto cfg = engines::YoloWorldOnnxConfig::from_env();
|
||||
detector = std::make_unique<engines::YoloWorldOnnx>(cfg);
|
||||
} else {
|
||||
auto cfg = engines::DetectionConfig::from_env();
|
||||
detector = std::make_unique<engines::ZedNative>(cfg);
|
||||
@@ -105,20 +101,6 @@ int main() {
|
||||
continue;
|
||||
}
|
||||
|
||||
auto input = event_as_input(std::move(event));
|
||||
std::string input_id = std::string(input.id);
|
||||
|
||||
if (input_id == "update_vocabulary") {
|
||||
if (!input.data.empty()) {
|
||||
std::string labels(
|
||||
reinterpret_cast<const char*>(input.data.data()),
|
||||
input.data.size());
|
||||
detector->set_labels(labels);
|
||||
std::cout << "Vocabulary updated: " << labels << std::endl;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!camera.grab()) {
|
||||
std::cerr << "Grab failed; attempting to reconnect..." << std::endl;
|
||||
camera.close();
|
||||
|
||||
Reference in New Issue
Block a user