Function main

software/example/sw_layer_plugin.cpp:56–308

using namespace cv;

int main(int argc, char **argv)
{
	if(argc < 5) {
		std::cout << "Usage : ./sw_layer_plugin.elf <quant_mode> <bit_width> <img1_path> <img2_path>" << std::endl;
		std::cout << "quant_mode : DynamicFixed" << std::endl;		
		std::cout << "bit_width : 6" << std::endl;
		std::cout << "img1_path : Path to the first image" << std::endl;
		std::cout << "img2_path : Path to the second image" << std::endl;
		return -1;			
	}

	string quant_mode_arg(argv[1]);
	int bit_width_arg   = atoi(argv[2]);
	char *img_path1 	= argv[3];
	char *img_path2 	= argv[4];
	
	char *dirpath; 
	char *caffemodel;
	char *prototxt;
	
	int resize_h = 224;
	int resize_w = 224;
	bool inp_mode;
	
	// Xilinx quantmode 6 bit
	if(quant_mode_arg == "DynamicFixed" && bit_width_arg == 6) {
		dirpath    = "models/GoogleNetWithoutLRN/6Bit";
		caffemodel = "GoogleNetNoLRN_6Bit_CHaiDNN.caffemodel";
		prototxt   = "GoogleNetNoLRN_6Bit_CHaiDNN_Customlayer.prototxt";
        inp_mode   = 1;
	}
	else {
		std::cout << "Usage : ./custom_ex.elf <quant_mode> <bit_width> <img1_path> <img2_path>" << std::endl;
		std::cout << "quant_mode : DynamicFixed" << std::endl;		
		std::cout << "bit_width : 6" << std::endl;
		std::cout << "img1_path : Path to the first image" << std::endl;
		std::cout << "img2_path : Path to the second image" << std::endl;
		return -1;
	}
	
	std :: cout << "[INFOx] Network Path : " << dirpath << std :: endl;

	//# start layer in the graph
	string start_layer = "";

	//# end layer in the graph
	string end_layer   = "";

	//# Flag indicate layer1 or not
	//# 1 - if image data is the for first layer input otherwise 0
	bool is_first_layer = 1;

	//# Number of images to process
	//# Supported batch size is 2
	int numImg_to_process = 2;

	//# Struct which holds the input/output layer info
	io_layer_info io_layer_info_ptr;