jacobgil / pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
https://jacobgil.github.io/pytorch-gradcam-book
MIT License
9.79k stars 1.52k forks source link

YOLOv5 2021-10-21 torch 2.0.1+cpu CPU Fusing layers... Model Summary: 213 layers, 7225885 parameters, 0 gradients Adding AutoShape... Traceback (most recent call last): File "E:\yolov5-v6.1\yolov5热力图.py", line 70, in <module> print(model.model.model.model[-2]) File "D:\anaconda\envs\fastapi\lib\site-packages\torch\nn\modules\module.py", line 1614, in __getattr__ raise AttributeError("'{}' object has no attribute '{}'".format( AttributeError: 'Sequential' object has no attribute 'model' #496

Open yqj0919 opened 3 months ago