// pch.h: 这是预编译标头文件。
// 下方列出的文件仅编译一次，提高了将来生成的生成性能。
// 这还将影响 IntelliSense 性能，包括代码完成和许多代码浏览功能。
// 但是，如果此处列出的文件中的任何一个在生成之间有更新，它们全部都将被重新编译。
// 请勿在此处添加要频繁更新的文件，这将使得性能优势无效。

#ifndef PCH_H
#define PCH_H

// 添加要在此处预编译的标头
#include "framework.h"
#endif //PCH_H

//定义宏
#ifdef IMPORT_DLL
#else
#define IMPORT_DLL extern "C" _declspec(dllimport)
#endif


IMPORT_DLL void* Init(
	const char* trt_file_path,
	int src_w,
	int src_h,
	float conf_thresh,
	float iou_thresh,
	int num_class
);
IMPORT_DLL void Detect(void* yolo, int rows, int cols, unsigned char* src_data, float(*res_array)[6]);