SparseConvNet/SpatiallySparseDatasetSHREC2015.cpp at master · SpiderMonkey1975/SparseConvNet · GitHub

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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
//Try adding normal vectors to input data.

//Data from http://www.icst.pku.edu.cn/zlian/shrec15-non-rigid/index.htm
//50 classes, 24 exemplars per class: alien ants armadillo bird1 bird2 camel cat centaur twoballs dinosaur dog1 dog2 glasses gorilla hand horse lamp paper man octopus pliers rabbit santa scissor shark snake spider dino_ske flamingo woman Aligator Bull Chick Deer Dragon Elephant Frog Giraffe Kangaroo Mermaid Mouse Nunchaku MantaRay Ring Robot Sumotori Tortoise Watch Weedle Woodman

#include "SpatiallySparseDatasetSHREC2015.h"
#include <vector>
#include <iostream>
#include <fstream>
#include <string>
#include "OpenCVPicture.h"
#include "Off3DFormatPicture.h"


SpatiallySparseDataset SHREC2015TrainSet(int renderSize,  int kFold, int fold) {
  SpatiallySparseDataset dataset;
  dataset.name="SHREC2015-Non-Rigid (Train subset)";
  dataset.type=TRAINBATCH;
  dataset.nFeatures=1;
  dataset.nClasses=50;
  std::ifstream cla("Data/SHREC15/SHREC15_Non-rigid_ToolKit/test.cla");
  std::string line;
  int nClasses, nTotal;
  getline(cla,line); //header line
  cla >> nClasses >> nTotal;
  for (int cl=0; cl<nClasses; cl++) {
    getline(cla,line); //blank line
    std::string className;
    int parent, nExemplars;
    cla >> className >> parent >> nExemplars;
    for (int exemp=0;exemp<nExemplars;exemp++) {
      int num;
      cla >> num;
      std::string filename=std::string("Data/SHREC15/SHREC15NonRigidTestDB/T")+std::to_string(num)+std::string(".off");
      if (exemp%kFold!=fold)
        dataset.pictures.push_back(new OffSurfaceModelPicture(filename, renderSize, cl));
    }
  }
  return dataset;
};

SpatiallySparseDataset SHREC2015TestSet(int renderSize,  int kFold, int fold) {
  SpatiallySparseDataset dataset;
  dataset.name="SHREC2015-Non-Rigid (Validation subset)";
  dataset.type=TESTBATCH;
  dataset.nFeatures=1;
  dataset.nClasses=50;
  std::ifstream cla("Data/SHREC15/SHREC15_Non-rigid_ToolKit/test.cla");
  std::string line;
  int nClasses, nTotal;
  getline(cla,line); //header line
  cla >> nClasses >> nTotal;
  for (int cl=0; cl<nClasses; cl++) {
    getline(cla,line); //blank line
    std::string className;
    int parent, nExemplars;
    cla >> className >> parent >> nExemplars;
    for (int exemp=0;exemp<nExemplars;exemp++) {
      int num;
      cla >> num;
      std::string filename=std::string("Data/SHREC15/SHREC15NonRigidTestDB/T")+std::to_string(num)+std::string(".off");
      if (exemp%kFold==fold)
        dataset.pictures.push_back(new OffSurfaceModelPicture(filename, renderSize, cl));
    }
  }
  return dataset;
};