Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,019 Commits 18 Branches 0 Tags
b5de52c4cdfefb3c7acfbff7d7f450a46b4aaada
Commit Graph
1019 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher b5de52c4cd torch.cuda.amp bug fix (#2750) 
PR https://github.com/ultralytics/yolov5/pull/2725 introduced a very specific bug that only affects multi-GPU trainings. Apparently the cause was using the torch.cuda.amp decorator in the autoShape forward method. I've implemented amp more traditionally in this PR, and the bug is resolved.
 2021-04-09 18:19:49 +02:00
Glenn Jocher fca5e2a48f autocast enable=torch.cuda.is_available() (#2748) 2021-04-09 13:34:49 +02:00
Glenn Jocher c03d590320 Add Hub results.pandas() method (#2725) 
* Add Hub results.pandas() method

New method converts results from torch tensors to pandas DataFrames with column names.

This PR may partially resolve issue https://github.com/ultralytics/yolov5/issues/2703

```python
results = model(imgs)

print(results.pandas().xyxy[0])
         xmin        ymin        xmax        ymax  confidence  class    name
0   57.068970  391.770599  241.383545  905.797852    0.868964      0  person
1  667.661255  399.303589  810.000000  881.396667    0.851888      0  person
2  222.878387  414.774231  343.804474  857.825073    0.838376      0  person
3    4.205386  234.447678  803.739136  750.023376    0.658006      5     bus
4    0.000000  550.596008   76.681190  878.669922    0.450596      0  person
```

* Update comments 

torch example input now shown resized to size=640 and also now a multiple of P6 stride 64 (see https://github.com/ultralytics/yolov5/issues/2722#issuecomment-814785930)

* apply decorators

* PEP8

* Update common.py

* pd.options.display.max_columns = 10

* Update common.py
 2021-04-07 16:28:07 +02:00
Glenn Jocher c8c8da6079 Update README with collapsable notes (#2721) 
* Update README with collapsable notes.

* cleanup

* center table
 2021-04-06 17:54:47 +02:00
Ayush Chaurasia 3067429307 Add support for list-of-directory data format for wandb (#2719) 2021-04-06 16:57:13 +02:00
Glenn Jocher ec8979f1d2 Updated filename attributes for YOLOv5 Hub BytesIO (#2718) 
Fix 2 for 'Model predict with forward will fail if PIL image does not have filename attribute' #2702
 2021-04-06 13:18:56 +02:00
Glenn Jocher 74276d5189 Updated filename attributes for YOLOv5 Hub results (#2708) 
Proposed fix for 'Model predict with forward will fail if PIL image does not have filename attribute' #2702
 2021-04-05 22:20:09 +02:00
Glenn Jocher 9ccfa85249 pip install coremltools onnx (#2690) 
Requested in https://github.com/ultralytics/yolov5/issues/2686
 2021-04-02 13:00:46 +02:00
Glenn Jocher 17300a4c7b autoShape forward im = np.asarray(im) # to numpy (#2689) 
Slight speedup.
 2021-04-02 12:36:38 +02:00
Glenn Jocher 2af059c0d8 PyTorch Hub model.save() increment as runs/hub/exp (#2684) 
* PyTorch Hub model.save() increment as runs/hub/exp

This chane will align PyTorch Hub results saving with the existing unified results saving directory structure of

runs/
  /train
  /detect
  /test
  /hub
    /exp
    /exp2
    ...

* cleanup
 2021-04-02 11:55:10 +02:00
Ayush Chaurasia 514ebcdf33 Fix: #2674 (#2683) 
* Set resume flag to false

* Check existance of val dataset
 2021-04-02 11:54:50 +02:00
Ding YiweiDingYiweiGlenn Jocher
 		1148e2ea63 Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 
* yolotr

* transformer block

* Remove bias in Transformer

* Remove C3T

* Remove a deprecated class

* put the 2nd LayerNorm into the 2nd residual block

* move example model to models/hub, rename to -transformer

* Add module comments and TODOs

* Remove LN in Transformer

* Add comments for Transformer

* Solve the problem of MA with DDP

* cleanup

* cleanup find_unused_parameters

* PEP8 reformat

Co-authored-by: DingYiwei <846414640@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-04-01 17:26:53 +02:00
Glenn Jocher b8b862993d Update README with Tips for Best Results tutorial (#2682) 
* Update README with Tips for Best Results tutorial

* Update README.md
 2021-04-01 15:01:00 +02:00
Glenn Jocher 877b826e3a Created using Colaboratory 2021-04-01 11:31:27 +02:00
Glenn Jocher 51cc0962b5 Update README.md 2021-04-01 11:16:56 +02:00
Ayush Chaurasia 2a28ef374b Set resume flag to false (#2657) 2021-03-31 13:47:54 +02:00
Glenn Jocher 1b475c1797 Created using Colaboratory 2021-03-30 20:07:18 +02:00
Phat TranphattranGlenn Jocher
 		9c803f2f7e Add --label-smoothing eps argument to train.py (default 0.0) (#2344) 
* Add label smoothing option

* Correct data type

* add_log

* Remove log

* Add log

* Update loss.py

remove comment (too versbose)

Co-authored-by: phattran <phat.tranhoang@cyberlogitec.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-03-29 18:45:46 +02:00
Benjamin FineranandGlenn Jocher fd1679975b add option to disable half precision in test.py (#2507) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-03-29 17:15:26 +02:00
Youngjin ShinandGlenn Jocher 7cdc5165a1 Update requirements.txt (#2564) 
* Add opencv-contrib-python to requirements.txt

* Update requirements.txt

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-03-29 17:05:52 +02:00
Glenn Jocher 1b100cd53e Created using Colaboratory (#2649) 2021-03-29 16:43:37 +02:00
Glenn Jocher 866bc7d640 Speed profiling improvements (#2648) 
* Speed profiling improvements

* Update torch_utils.py

deepcopy() required to avoid adding elements to model.

* Update torch_utils.py
 2021-03-29 15:19:07 +02:00
zzttqu 1e8ab3f5f2 Add tqdm pbar.close() (#2644) 
When using tqdm, sometimes it can't print in one line and roll to next line.
 2021-03-29 12:21:25 +02:00
Glenn Jocher 2bf34f50fd PyTorch Hub amp.autocast() inference (#2641) 
I think this should help speed up CUDA inference, as currently models may be running in FP32 inference mode on CUDA devices unnecesarily.
 2021-03-28 20:23:40 +02:00
Glenn Jocher ee169834bd PyTorch Hub custom model to CUDA device fix (#2636) 
Fix for #2630 raised by @Pro100rus32
 2021-03-28 17:22:00 +02:00
Glenn Jocher 2e95cf3d79 Improve git_describe() fix 1 (#2635) 
Add stderr=subprocess.STDOUT to catch error messages.
 2021-03-28 17:09:06 +02:00
Ayush Chaurasia 518c09578e W&B resume ddp from run link fix (#2579) 
* W&B resume ddp from run link fix

* Native DDP W&B support for training, resuming
 2021-03-28 16:11:36 +02:00
Ayush Chaurasia dc51e80b00 Fix: evolve with wandb (#2634) 2021-03-28 16:09:35 +02:00
Glenn Jocher 6e8c5b7678 Improve git_describe() (#2633) 
Catch 'fatal: not a git repository' returns and return '' instead (observed in GCP Hub checks).
 2021-03-28 15:39:31 +02:00
Glenn Jocher 9b92d3ee76 FROM nvcr.io/nvidia/pytorch:21.03-py3 (#2623) 
Update Docker FROM nvcr.io/nvidia/pytorch:21.03-py3
 2021-03-27 18:35:53 +01:00
Glenn Jocher 2dfe32030a Remove conflicting nvidia-tensorboard package (#2622) 
Attempt to resolve tensorboard Docker error in https://github.com/ultralytics/yolov5/issues/2573
 2021-03-27 18:31:53 +01:00
Glenn Jocher 005d7a8c54 Update Detections() self.n comment (#2620) 
```python
        self.n = len(self.pred)  # number of images (batch size)
```
 2021-03-26 21:19:15 +01:00
Ayush Chaurasia 8f6e447729 Fix Indentation in test.py (#2614) 
* Fix Indentation in test.py

* CI fix

* Comply with PEP8: 80 characters per line
 2021-03-26 20:47:11 +01:00
Glenn Jocher a57f23d18b Update detections() self.t = tuple() (#2617) 
* Update detections() self.t = tuple()

Fix multiple results.print() bug.

* Update experimental.py

* Update yolo.py
 2021-03-26 13:23:11 +01:00
Glenn Jocher 0ff5aeca61 Create date_modified() (#2616) 
Updated device selection string with fallback for non-git directories.
```python
def select_device(device='', batch_size=None):
    # device = 'cpu' or '0' or '0,1,2,3'
    s = f'YOLOv5 🚀 {git_describe() or date_modified()} torch {torch.__version__} '  # string
...
```
 2021-03-26 12:52:47 +01:00
maxuppandMax Uppenkamp 196bf10603 Add '*.mpo' to supported image formats (#2615) 
Co-authored-by: Max Uppenkamp <max.uppenkamp@inform-software.com>
 2021-03-26 12:45:22 +01:00
Glenn Jocher 77415a42e5 Update git_describe() for remote dir usage (#2606) 2021-03-26 01:44:00 +01:00
Glenn Jocher fca16dc4b3 Remove Cython from requirements.txt (#2604) 
Cython should be a dependency of the remaining packages in requirements.txt, so should be installed anyway even if not a direct requirement.
 2021-03-25 21:48:25 +01:00
Glenn Jocher 3bb414890a resume.py typo (#2603) 2021-03-25 20:55:20 +01:00
Glenn Jocher d4456e43b2 Update segment2box() comment (#2600) 2021-03-25 15:12:49 +01:00
Glenn Jocher ad05e37d99 Save webcam results, add --nosave option (#2598) 
This updates the default detect.py behavior to automatically save all inference images/videos/webcams unless the new argument --nosave is used (python detect.py --nosave) or unless a list of streaming sources is passed (python detect.py --source streams.txt)
 2021-03-25 14:09:49 +01:00
Max Kolomeychenko 16206692f2 Supervisely Ecosystem (#2519) 
guide describes YOLOv5 apps collection in Supervisely Ecosystem
 2021-03-25 00:57:34 +01:00
Glenn Jocher 333ccc5b0f YOLOv5 PyTorch Hub models >> check_requirements() (#2592) 
Improved user-feedback following requirements auto-update.
 2021-03-24 17:51:39 +01:00
Glenn Jocher 75feeb797c YOLOv5 PyTorch Hub models >> check_requirements() (#2591) 
Prints 'Please restart runtime or rerun command for update to take effect.' following package auto-install to inform users to restart/rerun.
 2021-03-24 16:42:54 +01:00
Glenn Jocher 8ace1b1b99 YOLOv5 PyTorch Hub models >> check_requirements() (#2588) 
* YOLOv5 PyTorch Hub models >> check_requirements()

Update YOLOv5 PyTorch Hub requirements.txt path to cache path.

* Update hubconf.py
 2021-03-24 16:23:54 +01:00
Glenn Jocher 9f98201dd9 W&B DDP fix 2 (#2587) 
Revert unintentional change to test batch sizes caused by PR https://github.com/ultralytics/yolov5/pull/2125
 2021-03-24 15:43:32 +01:00
Glenn Jocher 2bcc89d762 YOLOv5 PyTorch Hub models >> check_requirements() (#2577) 
* Update hubconf.py with check_requirements()

Dependency checks have been missing from YOLOv5 PyTorch Hub model loading, causing errors in some cases when users are attempting to import hub models in unsupported environments. This should examine the YOLOv5 requirements.txt file and pip install any missing or version-conflict packages encountered. 

This is highly experimental (!), please let us know if this creates problems in your custom workflows.

* Update hubconf.py
 2021-03-24 15:42:00 +01:00
Glenn Jocher e5b0200cd2 Update tensorboard>=2.4.1 (#2576) 
* Update tensorboard>=2.4.1 

Update tensorboard version to attempt to address https://github.com/ultralytics/yolov5/issues/2573 (tensorboard logging fail in Docker image).

* cleanup
 2021-03-24 01:29:00 +01:00
Glenn Jocher 2b329b0945 Enhanced check_requirements() with auto-install (#2575) 
* Update check_requirements() with auto-install

This PR builds on an idea I had to automatically install missing dependencies rather than simply report an error message. 

YOLOv5 should now 1) display all dependency issues and not simply display the first missing dependency, and 2) attempt to install/update each missing/VersionConflict package.

* cleanup

* cleanup 2

* Check requirements.txt file exists

* cleanup 3
 2021-03-24 01:05:59 +01:00
Ayush Chaurasia 1bf9365280 W&B DDP fix (#2574) 2021-03-23 16:54:34 +01:00