Compile fix

6 years ago · 480b3eccb1
parent c1f184249e
commit 480b3eccb1
1 changed files with 2 additions and 0 deletions
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@ -1173,6 +1173,7 @@ void assisted_excitation_forward(convolutional_layer l, network_state state)

    if(0)   // visualize ground truth
    {
+#ifdef OPENCV
        for (b = 0; b < l.batch; ++b)
        {
            image img = float_to_image(l.out_w, l.out_h, 1, &g[l.out_w*l.out_h*b]);
@ -1187,6 +1188,7 @@ void assisted_excitation_forward(convolutional_layer l, network_state state)
            wait_key_cv(5);
        }
        wait_until_press_key_cv();
+#endif // OPENCV
    }

    free(g);