Skip to content

Y
yolov5
切换分支/标签
  1. 03 8月, 2022 1 次提交
  3. 02 8月, 2022 6 次提交
    • Glenn Jocher's avatar
      New `smart_resume()` (#8838) · 08c8c3e0
      由 Glenn Jocher 提交于 
      * New `smart_resume()`

* Update torch_utils.py

* Update torch_utils.py

* Update torch_utils.py

* fix
      08c8c3e0
    • Jackson Argo's avatar
      Fix missing attr model.model when loading custom yolov model (#8830) · 2e109099
      由 Jackson Argo 提交于 
      * Update hubconf.py

Loading a custom yolov model causes this line to fail. Adding a test to check if the model actually has a model.model field. With this check, I'm able to load the model no prob.

Loading model via

```py
    model = torch.hub.load(
        'ultralytics/yolov5', 'custom', 'models/frozen_backbone_coco_unlabeled_best.onnx',
        autoshape=True, force_reload=False
    )
```

Causes traceback:

```
Traceback (most recent call last):
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/flask/app.py", line 2077, in wsgi_app
    response = self.full_dispatch_request()
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/flask/app.py", line 1525, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/flask/app.py", line 1523, in full_dispatch_request
    rv = self.dispatch_request()
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/flask/app.py", line 1509, in dispatch_request
    return self.ensure_sync(self.view_functions[rule.endpoint])(**req.view_args)
  File "/Users/jackson/Documents/GitHub/w210-capstone/api/endpoints/predictions.py", line 26, in post_predictions
    yolov_predictions = predict_bounding_boxes_for_collection(collection_id)
  File "/Users/jackson/Documents/GitHub/w210-capstone/api/predictions/predict_bounding_boxes.py", line 43, in predict_bounding_boxes_for_collection
    model = torch.hub.load(
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/torch/hub.py", line 404, in load
    model = _load_local(repo_or_dir, model, *args, **kwargs)
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/torch/hub.py", line 433, in _load_local
    model = entry(*args, **kwargs)
  File "/Users/jackson/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py", line 72, in custom
    return _create(path, autoshape=autoshape, verbose=_verbose, device=device)
  File "/Users/jackson/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py", line 67, in _create
    raise Exception(s) from e
Exception: 'DetectMultiBackend' object has no attribute 'model'. Cache may be out of date, try `force_reload=True` or see https://github.com/ultralytics/yolov5/issues/36 for help.
Exception on /api/v1/predictions [POST]
Traceback (most recent call last):
  File "/Users/jackson/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py", line 58, in _create
    model.model.model[-1].inplace = False  # Detect.inplace=False for safe multithread inference
  File "/Users/jackson/Documents/GitHub/w210-capstone/.venv/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1185, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'DetectMultiBackend' object has no attribute 'model'
```

* Update hubconf.py

* Update common.py
Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      2e109099
    • pre-commit-ci[bot]'s avatar
      [pre-commit.ci] pre-commit suggestions (#8828) · b7635efb
      由 pre-commit-ci[bot] 提交于 
      updates:
- [github.com/asottile/pyupgrade: v2.34.0 → v2.37.3](https://github.com/asottile/pyupgrade/compare/v2.34.0...v2.37.3)
- [github.com/PyCQA/flake8: 4.0.1 → 5.0.2](https://github.com/PyCQA/flake8/compare/4.0.1...5.0.2)
Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
      b7635efb
    • Glenn Jocher's avatar
      RUN git clone instead of COPY to `/usr/src/app` (#8827) · ba140e56
      由 Glenn Jocher 提交于 
      Update
      ba140e56
    • Glenn Jocher's avatar
      Remove hook `torch.nan_to_num(x)` (#8826) · f3c78a38
      由 Glenn Jocher 提交于 
      * Remove hook `torch.nan_to_num(x)`

Observed erratic training behavior (green line) with the nan_to_num hook in classifier branch. I'm going to remove it from master.

* Update train.py
      f3c78a38
    • Glenn Jocher's avatar
      Update README.md with contributors.png (#8820) · 59578f27
      由 Glenn Jocher 提交于 
      * Update README.md with contributors.png

Replace dynamic svg from opencollective with static png for improved stability and lighter size (400kB vs 2MB).

@AyushExel

* Update README.md

* Update README.md

* Update README_cn.md

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ciCo-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
      59578f27
  5. 01 8月, 2022 8 次提交
  7. 31 7月, 2022 10 次提交
  9. 30 7月, 2022 1 次提交
  11. 29 7月, 2022 4 次提交
  13. 27 7月, 2022 3 次提交
  15. 26 7月, 2022 3 次提交
  17. 25 7月, 2022 3 次提交
  19. 24 7月, 2022 1 次提交