update, support latest numpy, update readme and add version info

samson-wang · Nov 16, 2023 · a36abfa · a36abfa
1 parent 17a9209
commit a36abfa
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -15,9 +15,14 @@ pip install cython_bbox
 
 ```
 from cython_bbox import bbox_overlaps
+import numpy as np
+
+gt = np.random.random((5, 4))
+dt = np.random.random((10, 4))
+
 overlaps = bbox_overlaps(
-        np.ascontiguousarray(dt, dtype=np.float32),
-        np.ascontiguousarray(gt, dtype=np.float32)
+        np.ascontiguousarray(dt, dtype=np.float),
+        np.ascontiguousarray(gt, dtype=np.float)
     )
 
 ```
diff --git a/setup.py b/setup.py
@@ -48,7 +48,7 @@ def finalize_options(self):
     install_requires=["Cython","numpy"],
     ext_modules=ext_modules,
     cmdclass={'build': build},
-    version = '0.1.3',
+    version = '0.1.5',
     description = 'Standalone cython_bbox',
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/src/cython_bbox.pyx b/src/cython_bbox.pyx
@@ -9,8 +9,10 @@ cimport cython
 import numpy as np
 cimport numpy as np
 
-DTYPE = np.float
-ctypedef np.float_t DTYPE_t
+DTYPE = float
+ctypedef np.float64_t DTYPE_t
+
+__version__ = "0.1.5"
 
 def bbox_overlaps(
         np.ndarray[DTYPE_t, ndim=2] boxes,