OmniSciDB
a5dc49c757
|
#include <TorchWrapper.h>
Public Attributes | |
bool | is_valid {false} |
int64_t | batch_size {-1} |
int64_t | raster_channels {-1} |
int64_t | raster_tile_width {-1} |
int64_t | raster_tile_height {-1} |
int64_t | stride {-1} |
std::vector< std::string > | class_labels |
Definition at line 46 of file TorchWrapper.h.
int64_t ModelInfo::batch_size {-1} |
Definition at line 48 of file TorchWrapper.h.
Referenced by get_model_info_from_json().
std::vector<std::string> ModelInfo::class_labels |
Definition at line 53 of file TorchWrapper.h.
Referenced by get_model_info_from_json(), process_detections(), and TorchWarmer::warmup_torch().
bool ModelInfo::is_valid {false} |
Definition at line 47 of file TorchWrapper.h.
Referenced by __attribute__(), and get_model_info_from_json().
int64_t ModelInfo::raster_channels {-1} |
Definition at line 49 of file TorchWrapper.h.
Referenced by get_model_info_from_json().
int64_t ModelInfo::raster_tile_height {-1} |
Definition at line 51 of file TorchWrapper.h.
Referenced by get_model_info_from_json().
int64_t ModelInfo::raster_tile_width {-1} |
Definition at line 50 of file TorchWrapper.h.
Referenced by get_model_info_from_json().
int64_t ModelInfo::stride {-1} |
Definition at line 52 of file TorchWrapper.h.
Referenced by get_model_info_from_json().