sh::Extractor Class Reference

#include <extractor.h>

Public Member Functions
	Extractor ()
void	run ()

Private Attributes
torch::jit::script::Module	resnet152
torch::jit::script::Module	resnext101
at::Tensor	bias
at::Tensor	weights
at::Tensor	kw_pca_mat
at::Tensor	kw_pca_mean_vec
std::string	out_dir { "output/" }
std::string	thumbs { "data/ITEC_w2vv/thumbs/" }
std::string	frames { "data/ITEC_w2vv/ITEC.keyframes.dataset" }

Constructor & Destructor Documentation

Extractor()

Extractor::Extractor

(

)

Here is the call graph for this function:

Member Function Documentation

run()

void Extractor::run

(

)

Here is the call graph for this function:

Member Data Documentation

bias

at::Tensor sh::Extractor::bias

private

frames

std::string sh::Extractor::frames { "data/ITEC_w2vv/ITEC.keyframes.dataset" }

private

kw_pca_mat

at::Tensor sh::Extractor::kw_pca_mat

private

kw_pca_mean_vec

at::Tensor sh::Extractor::kw_pca_mean_vec

private

out_dir

std::string sh::Extractor::out_dir { "output/" }

private

resnet152

torch::jit::script::Module sh::Extractor::resnet152

private

resnext101

torch::jit::script::Module sh::Extractor::resnext101

private

thumbs

std::string sh::Extractor::thumbs { "data/ITEC_w2vv/thumbs/" }

private

weights

at::Tensor sh::Extractor::weights

private

---

The documentation for this class was generated from the following files:

• src/extractor/extractor.h
• src/extractor/extractor.cpp