mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
complete get_avgpool_image
get_avgpool_image is declared in avgpool_layer.h while it is not defined in avgpool_layer.c.
This commit is contained in:
parent
f61f3171d9
commit
817106e531
@ -2,6 +2,14 @@
|
||||
#include "cuda.h"
|
||||
#include <stdio.h>
|
||||
|
||||
image get_avgpool_image(avgpool_layer l)
|
||||
{
|
||||
int h = l.out_h;
|
||||
int w = l.out_w;
|
||||
int c = l.c;
|
||||
return float_to_image(w, h, c, l.output);
|
||||
}
|
||||
|
||||
avgpool_layer make_avgpool_layer(int batch, int w, int h, int c)
|
||||
{
|
||||
fprintf(stderr, "avg %4d x%4d x%4d -> %4d\n", w, h, c, c);
|
||||
|
Loading…
Reference in New Issue
Block a user