1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#ifndef _FRUGALLY_DEEP_H
#define _FRUGALLY_DEEP_H

// NOTE: assumes frugally 0.7.8-p0 is used

# include <emmintrin.h>

#include "AbstractCNNClassifier.h"

#include <fdeep/fdeep.hpp>


class FrugallyDeep : public AbstractCNNFinder {

public:
    FrugallyDeep(std::string file);
    virtual ~FrugallyDeep();

	void predict(const BallCandidates::PatchYUVClassified& p, double meanBrightness);<--- Function 'predict' argument 1 names different: declaration 'p' definition 'patch'.

    double getRadius();
    Vector2d getCenter();

private:
	std::shared_ptr<fdeep::model> model;
 
    std::vector<fdeep::tensor5> result;

};

#endif