Skip to content

Commit

Permalink
Merge pull request #630 from vigsterkr/latent
Browse files Browse the repository at this point in the history 
Fix indentation to comply with shogun standard
  • Loading branch information
Soeren Sonnenburg committed Jul 8, 2012
2 parents 0746d4a + e2ce7e7 commit ee792cd
Show file tree
Hide file tree
Showing 7 changed files with 623 additions and 624 deletions.
290 changes: 145 additions & 145 deletions examples/undocumented/libshogun/classifier_latent_svm.cpp
View file
Expand Up @@ -14,182 +14,182 @@ using namespace shogun;

struct CBoundingBox : public CLatentData
{
CBoundingBox(int32_t x, int32_t y) : CLatentData(), x_pos(x), y_pos(y) {};
CBoundingBox(int32_t x, int32_t y) : CLatentData(), x_pos(x), y_pos(y) {};

int32_t x_pos, y_pos;
int32_t x_pos, y_pos;

/** @return name of SGSerializable */
virtual const char* get_name() const { return "BoundingBox"; }
/** @return name of SGSerializable */
virtual const char* get_name() const { return "BoundingBox"; }
};

struct CHOGFeatures : public CLatentData
{
CHOGFeatures(int32_t w, int32_t h) : CLatentData(), width(w), height(h) {};
CHOGFeatures(int32_t w, int32_t h) : CLatentData(), width(w), height(h) {};

int32_t width, height;
float64_t ***hog;
int32_t width, height;
float64_t ***hog;

/** @return name of SGSerializable */
virtual const char* get_name() const { return "HOGFeatures"; }
/** @return name of SGSerializable */
virtual const char* get_name() const { return "HOGFeatures"; }
};

static void psi_hog(CLatentLinearMachine& llm, CLatentData* f, CLatentData* l, float64_t* psi)
{
CHOGFeatures* hf = (CHOGFeatures*) f;
CBoundingBox* bb = (CBoundingBox*) l;
for (int i = 0; i < llm.get_psi_size(); ++i)
{
psi[i] = hf->hog[bb->x_pos][bb->y_pos][i];
}
CHOGFeatures* hf = (CHOGFeatures*) f;
CBoundingBox* bb = (CBoundingBox*) l;
for (int i = 0; i < llm.get_psi_size(); ++i)
{
psi[i] = hf->hog[bb->x_pos][bb->y_pos][i];
}
}

static CLatentData* infer_latent_variable(CLatentLinearMachine& llm, CLatentData* f)
{
int32_t pos_x = 0, pos_y = 0;
int32_t w_dim;
float64_t max_score;

SGVector<float64_t> w = llm.get_w();
CHOGFeatures* hf = dynamic_cast<CHOGFeatures*> (f);
for (int i = 0; i < hf->width; ++i)
{
for (int j = 0; j < hf->height; ++j)
{
float64_t score = w.dot(w.vector, hf->hog[i][j], w.vlen);

if (score > max_score)
{
pos_x = i;
pos_y = j;
max_score = score;
}
}
}
SG_SDEBUG("%d %d %f\n", pos_x, pos_y, max_score);
CBoundingBox* h = new CBoundingBox(pos_x, pos_y);
SG_REF(h);

return h;
int32_t pos_x = 0, pos_y = 0;
int32_t w_dim;
float64_t max_score;

SGVector<float64_t> w = llm.get_w();
CHOGFeatures* hf = dynamic_cast<CHOGFeatures*> (f);
for (int i = 0; i < hf->width; ++i)
{
for (int j = 0; j < hf->height; ++j)
{
float64_t score = w.dot(w.vector, hf->hog[i][j], w.vlen);

if (score > max_score)
{
pos_x = i;
pos_y = j;
max_score = score;
}
}
}
SG_SDEBUG("%d %d %f\n", pos_x, pos_y, max_score);
CBoundingBox* h = new CBoundingBox(pos_x, pos_y);
SG_REF(h);

return h;
}

static void read_dataset(char* fname, CLatentFeatures*& feats, CLatentLabels*& labels)
{
FILE* fd = fopen(fname, "r");
char line[MAX_LINE_LENGTH];
char *pchar, *last_pchar;
int num_examples,label,height,width;

char* path = dirname(fname);

if (fd == NULL)
SG_SERROR("Cannot open input file %s!\n", fname);

fgets(line, MAX_LINE_LENGTH, fd);
num_examples = atoi(line);

labels = new CLatentLabels(num_examples);
SG_REF(labels);

feats = new CLatentFeatures(num_examples);
SG_REF(feats);

CMath::init_random();
for (int i = 0; (!feof(fd)) && (i < num_examples); ++i)
{
fgets(line, MAX_LINE_LENGTH, fd);

pchar = line;
while ((*pchar)!=' ') pchar++;
*pchar = '\0';
pchar++;

/* label: {-1, 1} */
last_pchar = pchar;
while ((*pchar)!=' ') pchar++;
*pchar = '\0';
label = (atoi(last_pchar) % 2 == 0) ? 1 : -1;
pchar++;

if (labels->set_label(i, label) == false)
SG_SERROR("Couldn't set label for element %d\n", i);

last_pchar = pchar;
while ((*pchar)!=' ') pchar++;
*pchar = '\0';
width = atoi(last_pchar);
pchar++;

last_pchar = pchar;
while ((*pchar)!='\n') pchar++;
*pchar = '\0';
height = atoi(last_pchar);

/* create latent label */
int x = CMath::random(0, width-1);
int y = CMath::random(0, height-1);
CBoundingBox* bb = new CBoundingBox(x,y);
labels->add_latent_label(bb);

SG_SPROGRESS(i, 0, num_examples);
CHOGFeatures* hog = new CHOGFeatures(width, height);
hog->hog = SG_CALLOC(float64_t**, hog->width);
for (int j = 0; j < width; ++j)
{
hog->hog[j] = SG_CALLOC(float64_t*, hog->height);
for (int k = 0; k < height; ++k)
{
char filename[MAX_LINE_LENGTH];
hog->hog[j][k] = SG_CALLOC(float64_t, HOG_SIZE);

sprintf(filename,"%s/%s.%03d.%03d.txt",path,line,j,k);
FILE* f = fopen(filename, "r");
if (f == NULL)
SG_SERROR("Could not open file: %s\n", filename);
for (int l = 0; l < HOG_SIZE; ++l)
fscanf(f,"%lf",&hog->hog[j][k][l]);
fclose(f);
}
}
feats->add_sample(hog);
}
fclose(fd);

SG_SDONE();
FILE* fd = fopen(fname, "r");
char line[MAX_LINE_LENGTH];
char *pchar, *last_pchar;
int num_examples,label,height,width;

char* path = dirname(fname);

if (fd == NULL)
SG_SERROR("Cannot open input file %s!\n", fname);

fgets(line, MAX_LINE_LENGTH, fd);
num_examples = atoi(line);

labels = new CLatentLabels(num_examples);
SG_REF(labels);

feats = new CLatentFeatures(num_examples);
SG_REF(feats);

CMath::init_random();
for (int i = 0; (!feof(fd)) && (i < num_examples); ++i)
{
fgets(line, MAX_LINE_LENGTH, fd);

pchar = line;
while ((*pchar)!=' ') pchar++;
*pchar = '\0';
pchar++;

/* label: {-1, 1} */
last_pchar = pchar;
while ((*pchar)!=' ') pchar++;
*pchar = '\0';
label = (atoi(last_pchar) % 2 == 0) ? 1 : -1;
pchar++;

if (labels->set_label(i, label) == false)
SG_SERROR("Couldn't set label for element %d\n", i);

last_pchar = pchar;
while ((*pchar)!=' ') pchar++;
*pchar = '\0';
width = atoi(last_pchar);
pchar++;

last_pchar = pchar;
while ((*pchar)!='\n') pchar++;
*pchar = '\0';
height = atoi(last_pchar);

/* create latent label */
int x = CMath::random(0, width-1);
int y = CMath::random(0, height-1);
CBoundingBox* bb = new CBoundingBox(x,y);
labels->add_latent_label(bb);

SG_SPROGRESS(i, 0, num_examples);
CHOGFeatures* hog = new CHOGFeatures(width, height);
hog->hog = SG_CALLOC(float64_t**, hog->width);
for (int j = 0; j < width; ++j)
{
hog->hog[j] = SG_CALLOC(float64_t*, hog->height);
for (int k = 0; k < height; ++k)
{
char filename[MAX_LINE_LENGTH];
hog->hog[j][k] = SG_CALLOC(float64_t, HOG_SIZE);

sprintf(filename,"%s/%s.%03d.%03d.txt",path,line,j,k);
FILE* f = fopen(filename, "r");
if (f == NULL)
SG_SERROR("Could not open file: %s\n", filename);
for (int l = 0; l < HOG_SIZE; ++l)
fscanf(f,"%lf",&hog->hog[j][k][l]);
fclose(f);
}
}
feats->add_sample(hog);
}
fclose(fd);

SG_SDONE();
}

int main(int argc, char** argv)
{
init_shogun_with_defaults();
sg_io->set_loglevel(MSG_DEBUG);
init_shogun_with_defaults();
sg_io->set_loglevel(MSG_DEBUG);

/* check whether the train/test args are given */
if (argc < 3)
{
SG_SERROR("not enough arguements given\n");
}
/* check whether the train/test args are given */
if (argc < 3)
{
SG_SERROR("not enough arguements given\n");
}

CLatentFeatures* train_feats = NULL;
CLatentLabels* train_labels = NULL;
/* read train data set */
read_dataset(argv[1], train_feats, train_labels);
CLatentFeatures* train_feats = NULL;
CLatentLabels* train_labels = NULL;
/* read train data set */
read_dataset(argv[1], train_feats, train_labels);

/* train the classifier */
float64_t C = 10.0;
/* train the classifier */
float64_t C = 10.0;

CLatentLinearMachine llm(C, train_feats, train_labels, HOG_SIZE);
llm.set_psi(psi_hog);
llm.set_infer(infer_latent_variable);
llm.train();
CLatentLinearMachine llm(C, train_feats, train_labels, HOG_SIZE);
llm.set_psi(psi_hog);
llm.set_infer(infer_latent_variable);
llm.train();

// CLatentFeatures* test_feats = NULL;
// CLatentLabels* test_labels = NULL;
// read_dataset(argv[2], test_feats, test_labels);
// CLatentFeatures* test_feats = NULL;
// CLatentLabels* test_labels = NULL;
// read_dataset(argv[2], test_feats, test_labels);

SG_SPRINT("Testing with the test set\n");
llm.apply(train_feats);
SG_SPRINT("Testing with the test set\n");
llm.apply(train_feats);


exit_shogun();
return 0;
exit_shogun();
return 0;
}

0 comments on commit ee792cd

Please sign in to comment.