diff --git a/darkflow/cython_utils/cy_yolo2_findboxes.pyx b/darkflow/cython_utils/cy_yolo2_findboxes.pyx index 299add00b..ee97c89aa 100644 --- a/darkflow/cython_utils/cy_yolo2_findboxes.pyx +++ b/darkflow/cython_utils/cy_yolo2_findboxes.pyx @@ -4,7 +4,7 @@ cimport cython ctypedef np.float_t DTYPE_t from libc.math cimport exp from ..utils.box import BoundBox -from nms cimport NMS +from darkflow.cython_utils.nms cimport NMS #expit @cython.boundscheck(False) # turn off bounds-checking for entire function diff --git a/darkflow/cython_utils/cy_yolo_findboxes.pyx b/darkflow/cython_utils/cy_yolo_findboxes.pyx index 1f654082f..2328ce4c5 100644 --- a/darkflow/cython_utils/cy_yolo_findboxes.pyx +++ b/darkflow/cython_utils/cy_yolo_findboxes.pyx @@ -4,7 +4,7 @@ cimport cython ctypedef np.float_t DTYPE_t from libc.math cimport exp from ..utils.box import BoundBox -from nms cimport NMS +from darkflow.cython_utils.nms cimport NMS