Just for display
| 27 | |
| 28 | //# Just for display |
| 29 | const char* getLayerType(kernel_type_e type) |
| 30 | { |
| 31 | switch (type) |
| 32 | { |
| 33 | case CONV: return "Convolution"; |
| 34 | case POOL: return "Pool"; |
| 35 | case FC_LAYER: return "FC"; |
| 36 | case SOFTMAX: return "SoftMax"; |
| 37 | case DECONV: return "Deconv"; |
| 38 | case NORM: return "Normalization"; |
| 39 | case NMS: return "NMS"; |
| 40 | case PERMUTE: return "Permute"; |
| 41 | case CROP: return "Crop"; |
| 42 | case XCUSTOM: return "Custom"; |
| 43 | case XPACK: return "Pack"; |
| 44 | } |
| 45 | return "Invalid Layer"; |
| 46 | } |
| 47 | |
| 48 | void *xiInit(char *dirpath, char* prototxt, char* caffemodel, |
| 49 | _io_layer_info *io_layer_info_ptr, int numImg_to_process, bool is_first_layer_in, std::string start_layer, std::string end_layer) |