OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

minor fix

This commit is contained in:
AlexeyAB 2020-11-18 04:06:28 +03:00
parent 17cf4bfbfc
commit 5c046bea96
3 changed files with 1430 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
build/darknet/x64/cfg/yolov4x-mish.cfg
View File

@ -1131,6 +1131,7 @@ size=1
stride=1 stride=1
pad=1 pad=1
activation=mish activation=mish
stopbackward=800
########################## ##########################

1428
cfg/yolov4x-mish.cfg Normal file
View File

File diff suppressed because it is too large Load Diff

2
src/yolo_layer.c
View File

@ -947,7 +947,7 @@ void forward_yolo_layer_gpu(const layer l, network_state state)
// if(y->1) x -> inf // if(y->1) x -> inf
// if(y->0) x -> -inf // if(y->0) x -> -inf
if (l.new_coords) { if (l.new_coords) {
activate_array_ongpu(l.output_gpu + index, 4 * l.w*l.h, LOGISTIC); // x,y activate_array_ongpu(l.output_gpu + index, 4 * l.w*l.h, LOGISTIC); // x,y,w,h
} }
else { else {
activate_array_ongpu(l.output_gpu + index, 2 * l.w*l.h, LOGISTIC); // x,y activate_array_ongpu(l.output_gpu + index, 2 * l.w*l.h, LOGISTIC); // x,y