yolo roi binary fix
This commit is contained in:
@@ -414,6 +414,11 @@ std::vector<fyt::auto_aim::Armor> ArmorYoloDetectorNode::detectArmors(
|
||||
// Convert ROS img to cv::Mat
|
||||
auto img = cv_bridge::toCvShare(img_msg, "bgr8")->image;
|
||||
|
||||
// Enable debug on detector if needed (before detect() so binary img gets stored)
|
||||
if (debug_ && debug_result_img_) {
|
||||
detector_->enable_debug = true;
|
||||
}
|
||||
|
||||
// Run detection: YOLO for ROI + traditional vision for refinement
|
||||
auto armors = detector_->detect(img);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user