/
Test.cxx
executable file
·31 lines (24 loc) · 902 Bytes
/
Test.cxx
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
#include <vtkSmartPointer.h>
#include <vtkJPEGReader.h>
#include <vtkJPEGWriter.h>
#include "vtkSuperPixelSegmentation.h"
int main(int argc, char* argv[])
{
vtkSmartPointer<vtkJPEGReader> reader =
vtkSmartPointer<vtkJPEGReader>::New();
reader->SetFileName(argv[1]);
reader->Update();
vtkSmartPointer<vtkSuperPixelSegmentation> superPixelSegmentation =
vtkSmartPointer<vtkSuperPixelSegmentation>::New();
superPixelSegmentation->SetSigma(.5);
superPixelSegmentation->SetK(500);
superPixelSegmentation->SetMinSize(50);
superPixelSegmentation->SetInputConnection(reader->GetOutputPort());
superPixelSegmentation->Update();
vtkSmartPointer<vtkJPEGWriter> writer =
vtkSmartPointer<vtkJPEGWriter>::New();
writer->SetFileName(argv[2]);
writer->SetInputConnection(superPixelSegmentation->GetOutputPort());
writer->Write();
return EXIT_SUCCESS;
}