extractor.cpp File Reference

#include <filesystem>
#include <fstream>
#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include "Extractor.h"

Include dependency graph for extractor.cpp:

Functions
std::vector< float >	parse_float_vector (const std::string &filepath, size_t dim, size_t begin_offset=0)
std::vector< cv::Rect >	get_RoIs (int width, int height)
torch::Tensor	get_features (cv::Mat image, torch::jit::script::Module resnext101, torch::jit::script::Module resnet152, torch::Tensor weights, torch::Tensor bias, torch::Tensor kw_pca_mat, torch::Tensor kw_pca_mean_vec)

Function Documentation

get_features()

torch::Tensor get_features	(	cv::Mat	image,
		torch::jit::script::Module	resnext101,
		torch::jit::script::Module	resnet152,
		torch::Tensor	weights,
		torch::Tensor	bias,
		torch::Tensor	kw_pca_mat,
		torch::Tensor	kw_pca_mean_vec
	)

Here is the call graph for this function:

get_RoIs()

std::vector<cv::Rect> get_RoIs	(	int	width,
		int	height
	)

parse_float_vector()

std::vector<float> parse_float_vector	(	const std::string &	filepath,
		size_t	dim,
		size_t	begin_offset = 0
	)