1. grad-cam 설치
!pip install grad-cam
2. 파일 추가
https://github.com/open-mmlab/mmdetection/pull/7987
demo/vis_cam.py
mmdet/utils/det_cam_visualizer.py
3. assert 삭제
demo/vis_cam.py
assert args.no_norm_in_bbox is False, 'If not norm in bbox, the ' \'visualization result ' \'may not be reasonable.'
4. python 실행
!python /content/mmdetection/demo/vis_cam.py /content/drive/MyDrive/12.jpg /content/mmdetection/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py /content/drive/MyDrive/epoch_24.pth --no-norm-in-bbox --method gradcam --out-dir /content/sample_data
'딥러닝' 카테고리의 다른 글
Intra-class distance (0) | 2022.10.25 |
---|---|
[COCO] 데이터 확인 (1) | 2022.10.04 |
[MMdetection] AP for each class (0) | 2022.09.21 |
[MMdetection] samples_per_gpu (0) | 2022.09.16 |
convert_to_coco (0) | 2022.09.14 |