OmniSciDB  a5dc49c757
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
TorchWrapper.h File Reference
+ 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< Detectiondetect_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: