TorchWrapper.h File Reference

#include "QueryEngine/TableFunctions/SystemFunctions/os/Shared/CpuTimer.hpp"
#include "QueryEngine/TableFunctions/SystemFunctions/os/Shared/RasterInfo.h"
#include <string>
#include <vector>

Include dependency graph for TorchWrapper.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	Detection
struct	BoxDetection
struct	ModelInfo
class	TorchWarmer

Functions
ModelInfo	get_model_info_from_file (const std::string &filename)
std::vector< Detection >	detect_objects_in_tiled_raster (const std::string &model_path, const ModelInfo &model_info, const bool use_gpu, const int64_t device_num, std::vector< float > &raster_data, const RasterFormat_Namespace::RasterInfo &raster_info, const float min_confidence_threshold, const float iou_threshold, std::shared_ptr< CpuTimer > timer)

Function Documentation

std::vector<Detection> detect_objects_in_tiled_raster	(	const std::string &	model_path,
		const ModelInfo &	model_info,
		const bool	use_gpu,
		const int64_t	device_num,
		std::vector< float > &	raster_data,
		const RasterFormat_Namespace::RasterInfo &	raster_info,
		const float	min_confidence_threshold,
		const float	iou_threshold,
		std::shared_ptr< CpuTimer >	timer
	)

Referenced by tf_torch_raster_obj_detect__cpu_template().

Here is the caller graph for this function:

ModelInfo get_model_info_from_file

(

const std::string &

filename

)

Referenced by tf_torch_raster_obj_detect__cpu_template(), and TorchWarmer::warmup_torch().

Here is the caller graph for this function: