Searched refs:softmax_tree (Results 1 – 5 of 5) sorted by relevance
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/ |
H A D | softmax_layer.c | 15 void softmax_tree(float *input, int batch, int inputs, float temp, tree *hierarchy, float *output) in softmax_tree() function 59 if(l.softmax_tree){ in forward_softmax_layer() 62 for (i = 0; i < l.softmax_tree->groups; ++i) { in forward_softmax_layer() 63 int group_size = l.softmax_tree->group_size[i]; in forward_softmax_layer() 91 if(l.softmax_tree){ in forward_softmax_layer_gpu() 92 softmax_tree_gpu(net.input, 1, l.batch, l.inputs, l.temperature, l.output_gpu, *l.softmax_tree); in forward_softmax_layer_gpu() 111 if(l.softmax_tree){ in forward_softmax_layer_gpu()
|
H A D | region_layer.c | 181 void softmax_tree(float *input, int batch, int inputs, float temp, tree *hierarchy, float *output); 199 if (l.softmax_tree){ in forward_region_layer() 203 softmax_tree(l.output + index + 5, 1, 0, 1, l.softmax_tree, l.output + index + 5); in forward_region_layer() 226 if(l.softmax_tree){ in forward_region_layer() 238 … float p = scale*get_hierarchy_probability(l.output + index, l.softmax_tree, class_id); in forward_region_layer() 394 if(l.softmax_tree){ in get_region_boxes() 396 hierarchy_predictions(predictions + class_index, l.classes, l.softmax_tree, 0); in get_region_boxes() 438 if(l.softmax_tree){ in forward_region_layer_gpu() 441 for (i = 0; i < l.softmax_tree->groups; ++i) { in forward_region_layer_gpu() 442 int group_size = l.softmax_tree->group_size[i]; in forward_region_layer_gpu() [all …]
|
H A D | parser.c | 343 if (tree_file) layer.softmax_tree = read_tree(tree_file); in parse_softmax() 636 if (tree_file) l.softmax_tree = read_tree(tree_file); in parse_region() 1338 net.hierarchy = l.softmax_tree; in parse_network_cfg_custom()
|
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/darknet/ |
H A D | darknet_io.cpp | 443 … int coords, int classes, int anchors, int classfix, int softmax, int softmax_tree, float *biasDat… in setRegion() argument 454 region_param.set<bool>("softmax_tree", softmax_tree); in setRegion() 831 … bool softmax_tree = (getParam<std::string>(layer_params, "tree", "").size() > 0); in ReadDarknetFromCfgStream() local 840 …setRegion(thresh, coords, classes, num_of_anchors, classfix, softmax, softmax_tree, anchors_vec.da… in ReadDarknetFromCfgStream()
|
/dports/misc/darknet/darknet-darknet_yolo_v4_pre/include/ |
H A D | darknet.h | 510 tree *softmax_tree; member
|