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