/gradcam_features_yolo

Visualizing Yolov5's layers using GradCam

Primary LanguagePythonMIT LicenseMIT

run

 python main.py --model-path last.pt --img-path images/Frame_1.JPG --output-dir out --target-layer model_9_cv2_act --min_plots 64