mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
another compile fix
This commit is contained in:
@ -23,7 +23,7 @@
|
|||||||
layer make_gaussian_yolo_layer(int batch, int w, int h, int n, int total, int *mask, int classes)
|
layer make_gaussian_yolo_layer(int batch, int w, int h, int n, int total, int *mask, int classes)
|
||||||
{
|
{
|
||||||
int i;
|
int i;
|
||||||
layer l = {0};
|
layer l = { (LAYER_TYPE)0 };
|
||||||
l.type = GAUSSIAN_YOLO;
|
l.type = GAUSSIAN_YOLO;
|
||||||
|
|
||||||
l.n = n;
|
l.n = n;
|
||||||
|
12
src/parser.c
12
src/parser.c
@ -159,9 +159,15 @@ convolutional_layer parse_convolutional(list *options, size_params params, netwo
|
|||||||
int n = option_find_int(options, "filters",1);
|
int n = option_find_int(options, "filters",1);
|
||||||
int groups = option_find_int_quiet(options, "groups", 1);
|
int groups = option_find_int_quiet(options, "groups", 1);
|
||||||
int size = option_find_int(options, "size",1);
|
int size = option_find_int(options, "size",1);
|
||||||
int stride = option_find_int(options, "stride",1);
|
int stride = -1;
|
||||||
int stride_x = option_find_int_quiet(options, "stride_x", stride);
|
//int stride = option_find_int(options, "stride",1);
|
||||||
int stride_y = option_find_int_quiet(options, "stride_y", stride);
|
int stride_x = option_find_int_quiet(options, "stride_x", -1);
|
||||||
|
int stride_y = option_find_int_quiet(options, "stride_y", -1);
|
||||||
|
if (stride_x < 1 || stride_y < 1) {
|
||||||
|
stride = option_find_int(options, "stride", 1);
|
||||||
|
if (stride_x < 1) stride_x = stride;
|
||||||
|
if (stride_y < 1) stride_y = stride;
|
||||||
|
}
|
||||||
int dilation = option_find_int_quiet(options, "dilation", 1);
|
int dilation = option_find_int_quiet(options, "dilation", 1);
|
||||||
int antialiasing = option_find_int_quiet(options, "antialiasing", 0);
|
int antialiasing = option_find_int_quiet(options, "antialiasing", 0);
|
||||||
if (size == 1) dilation = 1;
|
if (size == 1) dilation = 1;
|
||||||
|
Reference in New Issue
Block a user