lsgrmController.txx 10.37 KiB
#ifndef __LSGRM_CONTROLLER_TXX
#define __LSGRM_CONTROLLER_TXX
namespace lsgrm
	template<class TSegmenter>
	Controller<TSegmenter>::Controller()
		m_Memory = 0;
		m_ImageDivisionActivated = true;
	template<class TSegmenter>
	Controller<TSegmenter>::~Controller()
	template<class TSegmenter>
	void Controller<TSegmenter>::RunSegmentation()
		// Rajouter un if pour vérifier si l'utilisateur a enclenché la procédure automatique
		if(m_Memory < 1)
			this->GetAutomaticConfiguration();
			std::cout << m_Memory  << " bytes, tile dimension " << m_TileWidth << " X " << m_TileHeight
					  << ", margin" << m_Margin  << " niter " << m_NumberOfFirstIterations << std::endl;
		// Divide the input image if necessary
		if(m_ImageDivisionActivated)
			SplitOTBImage<ImageType>(m_InputImage, m_TileDirectory, m_TileWidth,
									 m_TileHeight, m_Margin, m_NumberOfFirstIterations);
		// Retrieve the problem configuration
		RetrieveProblemConfiguration();
		// Print values
		std::cout << m_Memory  << " bytes, tile dimension " << m_TileWidth << " X " << m_TileHeight
				  << ", margin" << m_Margin  << " niter " << m_NumberOfFirstIterations << std::endl;
		// Boolean indicating if there are remaining fusions
		bool isFusion = false;
		// Run first partial segmentation
		auto accumulatedMemory = RunFirstPartialSegmentation<TSegmenter>(m_SpecificParameters,
																		 m_Threshold,
																		 m_NumberOfFirstIterations,
																		 m_Tiles,
																		 m_TileDirectory,
																		 m_NbTilesX,
																		 m_NbTilesY,
																		 m_Margin,
																		 m_TileWidth,
																		 m_TileHeight,
																		 m_ImageWidth,
																		 m_ImageHeight,
																		 m_TemporaryDirectory,
																		 isFusion);
		std::cout << "Accumulated memory " << accumulatedMemory << " bytes, there is fusion "<< isFusion << std::endl;
		while(accumulatedMemory > m_Memory && isFusion)
		    isFusion = false;
			accumulatedMemory = RunPartialSegmentation<TSegmenter>(m_SpecificParameters,
																   m_Threshold,
																   m_Tiles,
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
m_TemporaryDirectory, m_NbTilesX, m_NbTilesY, m_TileWidth, m_TileHeight, m_ImageWidth, m_ImageHeight, m_ImageBands, isFusion); std::cout << "Accumulated memory " << accumulatedMemory << " bytes, there is fusion " << isFusion << std::endl; } if(accumulatedMemory <= m_Memory) { // Merge all the graphs MergeAllGraphsAndAchieveSegmentation<TSegmenter>(m_SpecificParameters, m_Threshold, m_Tiles, m_TemporaryDirectory, m_NbTilesX, m_NbTilesY, m_TileWidth, m_TileHeight, m_ImageWidth, m_ImageHeight, m_ImageBands, isFusion, m_OutputGraphDirectory); } else { // That means there are no more possible fusions but we can not store the ouput graph // Todo do not clean up temporary directory before copying resulting graph to the output directory // In the output directory add an info file to give the number of tiles. } } template<class TSegmenter> void Controller<TSegmenter>::RetrieveProblemConfiguration() { // Open the lsgrm info file std::ifstream in(m_TileDirectory + "info.txt"); assert(in.good()); std::string line; std::vector<std::string> tokens; std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_ImageWidth = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_ImageHeight = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_ImageBands = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_NbTilesX = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_NbTilesY = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line);
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
boost::split(tokens, line, boost::is_any_of(":")); m_TileWidth = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_TileHeight = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_Margin = static_cast<unsigned int>(atoi(tokens[1].c_str())); std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); m_NumberOfFirstIterations = static_cast<unsigned int>(atoi(tokens[1].c_str())); m_Tiles.assign(m_NbTilesX * m_NbTilesY, ProcessingTile()); std::vector<std::string> subtokens; unsigned int i = 0; for(unsigned int row = 0; row < m_NbTilesY; ++row) { for(unsigned int col = 0; col < m_NbTilesX; ++col) { std::getline(in, line); boost::split(tokens, line, boost::is_any_of(":")); boost::split(subtokens, tokens[1], boost::is_any_of(",")); /* Margin at the top ? */ if( row > 0 ) { m_Tiles[i].rows[0] = row * m_TileHeight; m_Tiles[i].tileNeighbors[0] = i - m_NbTilesX; m_Tiles[i].margin[0] = true; }else { m_Tiles[i].rows[0] = 0; m_Tiles[i].tileNeighbors[0] = -1; m_Tiles[i].margin[0] = false; } /* Margin at the right ? */ if( col < m_NbTilesX - 1 ) { m_Tiles[i].columns[1] = col * m_TileWidth + static_cast<unsigned int>(atoi(subtokens[2].c_str())) - 1; m_Tiles[i].tileNeighbors[2] = i+1; m_Tiles[i].margin[1] = true; } else { m_Tiles[i].columns[1] = m_ImageWidth - 1; m_Tiles[i].tileNeighbors[2] = -1; m_Tiles[i].margin[1] = false; } /* Margin at the bottom */ if( row < m_NbTilesY - 1) { m_Tiles[i].rows[1] = row * m_TileHeight + static_cast<unsigned int>(atoi(subtokens[3].c_str())) - 1; m_Tiles[i].tileNeighbors[4] = i + m_NbTilesX; m_Tiles[i].margin[2] = true; } else { m_Tiles[i].rows[1] = m_ImageHeight - 1; m_Tiles[i].tileNeighbors[4] = -1; m_Tiles[i].margin[2] = false; } /* Margin at the left */ if( col > 0 ) {
211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280
m_Tiles[i].columns[0] = col * m_TileWidth; m_Tiles[i].tileNeighbors[6] = i-1; m_Tiles[i].margin[3] = true; } else { m_Tiles[i].columns[0] = 0; m_Tiles[i].tileNeighbors[6] = -1; m_Tiles[i].margin[3] = false; } /* Is there a neighbor at the rop right */ if(row > 0 && col < m_NbTilesX - 1) m_Tiles[i].tileNeighbors[1] = i - m_NbTilesX + 1; else m_Tiles[i].tileNeighbors[1] = -1; /* Is there a neighbor at the bottom right */ if(col < m_NbTilesX - 1 && row < m_NbTilesY - 1) m_Tiles[i].tileNeighbors[3] = i + m_NbTilesX + 1; else m_Tiles[i].tileNeighbors[3] = -1; /* Is there a neighbor at the bottom left */ if(row < m_NbTilesY - 1 && col > 0) m_Tiles[i].tileNeighbors[5] = i + m_NbTilesX - 1; else m_Tiles[i].tileNeighbors[5] = -1; /* Is there a neighbor at the top left */ if(col > 0 && row > 0) m_Tiles[i].tileNeighbors[7] = i - m_NbTilesX - 1; else m_Tiles[i].tileNeighbors[7] = -1; i++; } } in.close(); } template<class TSegmenter> void Controller<TSegmenter>::GetAutomaticConfiguration() { m_Memory = getMemorySize(); assert(m_Memory > 0); m_Memory /= 2; // For safety and can prevent out of memory troubles // Compute the size of an initial segment using NodeType = typename TSegmenter::NodeType; using NodePointer = typename TSegmenter::NodePointerType; using EdgeType = typename TSegmenter::EdgeType; long long unsigned int sizePerNode = sizeof(NodePointer) + sizeof(NodeType) + 1 + 4 *(sizeof(EdgeType) + sizeof(float)); // last term is specific to BS. long unsigned int maximumNumberOfNodes = std::ceil(m_Memory / sizePerNode); unsigned int tileDimension = std::sqrt(maximumNumberOfNodes); // Compute the stability margin. The naive strategy consider a margin value and a stable size equal. unsigned int niter = 1; unsigned int maxMargin = tileDimension/2; unsigned int currMargin = static_cast<unsigned int>(pow(2, niter + 1) - 2); unsigned int prevMargin = currMargin; while(currMargin < maxMargin) { prevMargin = currMargin; niter++; currMargin = static_cast<unsigned int>(pow(2, niter + 1) - 2);
281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350
} m_TileWidth = tileDimension - prevMargin; m_TileHeight = m_TileWidth; m_Margin = prevMargin; m_NumberOfFirstIterations = niter - 1; } template <class TSegmenter> void Controller<TSegmenter>::SetInternalMemoryAvailable(long long unsigned int v) // expecting a value in Mbytes. { assert(v > 0); m_Memory = v * 1024ul * 1024ul; } template<class TSegmenter> void Controller<TSegmenter>::SetImageDivision(bool f) { m_ImageDivisionActivated = f; } template<class TSegmenter> void Controller<TSegmenter>::SetInputImage(const std::string& str) { m_InputImage = str; } template<class TSegmenter> void Controller<TSegmenter>::SetOutputGraphDirectory(const std::string& str) { m_OutputGraphDirectory = str; } template<class TSegmenter> void Controller<TSegmenter>::SetTileDirectory(const std::string& str) { m_TileDirectory = str; } template<class TSegmenter> void Controller<TSegmenter>::SetTemporaryDirectory(const std::string& str) { m_TemporaryDirectory = str; } template<class TSegmenter> void Controller<TSegmenter>::SetTileWidth(const unsigned int v) { m_TileWidth = v; } template<class TSegmenter> void Controller<TSegmenter>::SetTileHeight(const unsigned int v) { m_TileHeight = v; } template<class TSegmenter> void Controller<TSegmenter>::SetNumberOfFirstIterations(const unsigned int v) { m_NumberOfFirstIterations = v; m_Margin = static_cast<unsigned int>(pow(2, m_NumberOfFirstIterations + 1) - 2);// 2^{n+1}-2 } template<class TSegmenter> void Controller<TSegmenter>::SetSpecificParameters(const SegmentationParameterType& params) { m_SpecificParameters = params; }
351352353354355356357358359
template<class TSegmenter> void Controller<TSegmenter>::SetThreshold(const float& t) { m_Threshold = t; } } // end of namespace lsgrm #endif