Home
last modified time | relevance | path

Searched refs:softmax_tree (Results 1 – 5 of 5) sorted by relevance

/dports/misc/darknet/darknet-darknet_yolo_v4_pre/src/
H A Dsoftmax_layer.c15 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 Dregion_layer.c181 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 Dparser.c343 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 Ddarknet_io.cpp443 … 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 Ddarknet.h510 tree *softmax_tree; member