traffic_light_detection_component.cc

浏览该文件的文档.

/******************************************************************************
 * Copyright 2023 The Apollo Authors. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *****************************************************************************/
#include "modules/perception/traffic_light_detection/traffic_light_detection_component.h"
 
#include "modules/perception/traffic_light_detection/proto/traffic_light_detection_component.pb.h"
 
#include "cyber/profiler/profiler.h"
#include "cyber/time/clock.h"
#include "modules/perception/common/camera/common/trafficlight_frame.h"
#include "modules/perception/traffic_light_detection/interface/base_traffic_light_detector.h"
 
namespace apollo {
namespace perception {
namespace trafficlight {
 
bool TrafficLightDetectComponent::Init() {
    // init component config
    if (InitConfig() != cyber::SUCC) {
        AERROR << "TrafficLightDetectComponent InitConfig failed.";
        return false;
    }
    // init algorithm plugin
    if (InitAlgorithmPlugin() != cyber::SUCC) {
        AERROR << "TrafficLightDetectComponent InitAlgorithmPlugin failed.";
        return false;
    }
 
    return true;
}
 
bool TrafficLightDetectComponent::Proc(const std::shared_ptr<TrafficDetectMessage>& message) {
    PERF_FUNCTION()
    auto time_imags = std::to_string(message->timestamp_);
    AINFO << "Enter detection component, message timestamp: " << time_imags;
 
    bool status = InternalProc(message);
    if (status) {
        writer_->Write(message);
        AINFO << "Send trafficlight detect output message.";
    }
    return status;
}
 
int TrafficLightDetectComponent::InitConfig() {
    apollo::perception::trafficlight::TrafficLightParam traffic_light_param;
    if (!GetProtoConfig(&traffic_light_param)) {
        AINFO << "load trafficlights detection component proto param failed";
        return cyber::FAIL;
    }
 
    auto plugin_param = traffic_light_param.plugin_param();
    tl_detector_name_ = plugin_param.name();
    config_path_ = plugin_param.config_path();
    config_file_ = plugin_param.config_file();
 
    gpu_id_ = traffic_light_param.gpu_id();
    AINFO << "tl_detector_name: " << tl_detector_name_ << " config_path: " << config_path_
          << " config_file: " << config_file_ << " gpu_id: " << gpu_id_;
 
    writer_ = node_->CreateWriter<TrafficDetectMessage>(traffic_light_param.detection_output_channel_name());
 
    return cyber::SUCC;
}
 
bool TrafficLightDetectComponent::InitAlgorithmPlugin() {
    trafficlight::BaseTrafficLightDetector* detector
            = trafficlight::BaseTrafficLightDetectorRegisterer::GetInstanceByName(tl_detector_name_);
    CHECK_NOTNULL(detector);
    detector_.reset(detector);
    ACHECK(detector_ != nullptr);
    detector_init_options_.gpu_id = gpu_id_;
    detector_init_options_.config_path = config_path_;
    detector_init_options_.config_file = config_file_;
 
    if (!detector_->Init(detector_init_options_)) {
        AERROR << "Trafficlight detector init failed";
        return false;
    }
 
    return cyber::SUCC;
}
 
bool TrafficLightDetectComponent::InternalProc(const std::shared_ptr<TrafficDetectMessage const>& message) {
    PERF_BLOCK("traffic_light_detector")
    bool status = detector_->Detect(message->traffic_light_frame_.get());
    PERF_BLOCK_END
 
    return true;
}
 
}  // namespace trafficlight
}  // namespace perception
}  // namespace apollo