V0.4.7.4

Author: lipengbo

README.md

@@ -17,6 +17,7 @@ pip install git+https://github.com/bobo0810/bobotools.git
 |  ----  | ----  |
 | get_model_info  | 获取模型信息(模型大小、计算量、参数量、前向推理耗时等) |
 | tensor2img  | Tenso恢复为图像，方便可视化|
+| vis_cam  | 可视化注意力图|
 
 
 ### 图像(img_tools)

setup.py

@@ -1,11 +1,22 @@
 from setuptools import setup, find_packages
 import os
+
 ROOT_DIR = os.path.dirname(os.path.realpath(__file__))
 
 # f= open (os.path.join(ROOT_DIR, 'bobotools/requirements.txt'),'r')
 # install_requires = [line.strip()  for line in  f.readlines()]
 # f.close()
-install_requires=['pytest','tqdm','opencv-python-headless','numpy','ptflops','torch','torchvision','grad-cam','pillow']
+install_requires = [
+    "pytest",
+    "tqdm",
+    "opencv-python-headless",
+    "numpy",
+    "ptflops",
+    "torch",
+    "torchvision",
+    "grad-cam",
+    "pillow",
+]
 
 setup(
     name="bobotools",
@@ -15,7 +26,7 @@
     description="bobotools",
     long_description=open("README.md", "r").read(),
     long_description_content_type="text/markdown",
-    version="0.4.7.3",  # 版本
+    version="0.4.7.4",  # 版本
     install_requires=install_requires,
     python_requires=">=3.6",
     include_package_data=True,