208b2fcf39
* add simplified Logger class * add modified yolop tensorrtx implementation * add gen_wts.py * add CMakeLists.txt * add yolop tensorrtx README.md * fix yolop spelling * simplify yolop code * fix yolop README.md Co-authored-by: aliceint(zhanglj) <2464313547@qq.com>
25 lines
490 B
C++
25 lines
490 B
C++
// create by ausk(jinlj) 2022/10/25
|
|
#pragma once
|
|
|
|
#include "NvInferRuntimeCommon.h"
|
|
#include <cassert>
|
|
#include <ctime>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include <ostream>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
using Severity = nvinfer1::ILogger::Severity;
|
|
|
|
class Logger : public nvinfer1::ILogger
|
|
{
|
|
public:
|
|
void log(Severity severity, const char* msg) noexcept override
|
|
{
|
|
if (severity < Severity::kINFO) {
|
|
std::cout << msg << std::endl;
|
|
}
|
|
}
|
|
};
|