Typo in BaseModel._forward_once

[fcakyon]

Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Detection

Bug

save_dir is assigned with bool value which raises an error (it should be Path)

yolov5/models/yolo.py

Line 124 in 03f2ca8

feature_visualization(x, m.type, m.i, save_dir=visualize)

Environment

YOLOv5 6.2 (commit: 03f2ca8)
Ubuntu 18.04
Python 3.8

Minimal Reproducible Example

model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

result = model(Image.open('data/images/bus.jpg'), size=1344, augment=False, profile=True)

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[glenn-jocher]

@fcakyon good news 😃! Your original issue may now be fixed ✅ in PR #9443. This PR restricts PyTorch Hub models to simply using augment option.

You can visualize models with python detect.py --visualize and you can profile model yamls with python models/yolo.py --profile or --line-profile

To receive this update:

• Git – git pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
• PyTorch Hub – Force-reload model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
• Notebooks – View updated notebooks
• Docker – sudo docker pull ultralytics/yolov5:latest to update your image

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!