Update activation_layer.h

2021-12-10 08:24:01 -03:00
parent bfd9268a31
commit 7761ca7a6b
1 changed files with 1 additions and 0 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/activation_layer.h
@@ -8,6 +8,7 @@

 #include <string>
 #include <cassert>
+#include <iostream>

 #include "NvInfer.h"