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
3 years ago
d4456e43b2
Update segment2box() comment ( #2600 )
3 years ago
333ccc5b0f
YOLOv5 PyTorch Hub models >> check_requirements() ( #2592 )
Improved user-feedback following requirements auto-update.
3 years ago
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.
3 years ago
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
3 years ago
9b11f0c58b
PyTorch Hub models default to CUDA:0 if available ( #2472 )
* PyTorch Hub models default to CUDA:0 if available
* device as string bug fix
3 years ago
ba18528b47
bbox_iou() stability and speed improvements ( #2385 )
3 years ago
fab5085674
EMA bug fix 2 ( #2330 )
* EMA bug fix 2
* update
3 years ago
ec1d8496ba
Improved model+EMA checkpointing ( #2292 )
* Enhanced model+EMA checkpointing
* update
* bug fix
* bug fix 2
* always save optimizer
* ema half
* remove model.float()
* model half
* carry ema/model in fp32
* rm model.float()
* both to float always
* cleanup
* cleanup
3 years ago
c09964c27c
Update inference default to multi_label=False ( #2252 )
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
3 years ago
d2e754b67b
Add isdocker() ( #2232 )
* Add isdocker()
* Update general.py
* Update general.py
3 years ago
26c2e54c8f
Add check_imshow() ( #2231 )
* Add check_imshow()
* Update general.py
* Update general.py
3 years ago
bdd88e1ed7
YOLOv5 Segmentation Dataloader Updates ( #2188 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* GitHub API rate limit fix
* update
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* astuple
* epochs
* update
* update
* ComputeLoss()
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* merge
* merge
* merge
* merge
* update
* update
* update
* update
* commit=tag == tags[-1]
* Update cudnn.benchmark
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* mosaic9
* update
* update
* update
* update
* update
* update
* institute cache versioning
* only display on existing cache
* reverse cache exists booleans
3 years ago
a5d5f9262d
Changed socket port and added timeout ( #2176 )
3 years ago
f59f80114c
Add 'exclude' tuple to check_requirements() ( #2041 )
3 years ago
3b7feeafdf
Update general.py check_git_status() fix ( #2020 )
3 years ago
aac33f87ad
Update inference multiple-counting ( #2019 )
* Update inference multiple-counting
* update github check
3 years ago
1445ab2b27
check_git_status() Windows fix ( #2015 )
* check_git_status() Windows fix
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
3 years ago
d9212140b3
Add xywhn2xyxy() ( #1983 )
3 years ago
b1cf25dd9a
check_git_status() asserts ( #1977 )
3 years ago
b26a2f6242
check_git_status() when not exist /workspace ( #1966 )
3 years ago
e8a41e8916
prevent check_git_status() in docker images ( #1951 )
* prevent check_git_status() running docker images
* Update general.py
3 years ago
051e9e80de
check_git_status() bug fix ( #1925 )
3 years ago
509dd51aca
check_git_status() improvements ( #1916 )
* check_online()
* Update general.py
* update check_git_status()
* reverse rev-parse order
* fetch
* improved responsiveness
* comment
* comment
* remove hyp['giou'] compat warning
3 years ago
dd03b20ba5
colorstr() updates ( #1909 )
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
3 years ago
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
3 years ago
6ab589583c
Add colorstr() ( #1887 )
* Add colorful()
* update
* newline fix
* add git description
* --always
* update loss scaling
* update loss scaling 2
* rename to colorstr()
3 years ago
e77c77f580
Add check_requirements() ( #1853 )
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
3 years ago
135ec5c5ce
W&B ID reset on training completion ( #1852 )
* Update train.py
Fix the bug of always the same W&B ID and continue overwrite with the old logging.
BUG report
https://github.com/ultralytics/yolov5/issues/1851
* Fix the bug of duplicate W&B ID
fix the bug of https://github.com/ultralytics/yolov5/issues/1851
If we had trained on yolov5s.pt, the program will generate a new unique W&B ID.
If we hadn't, the program will keep the old code, we can still use --resume aug.
* Update general.py
* revert train.py changes
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
69be8e738f
YOLOv5 v4.0 Release ( #1837 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update study.png
* update study.png
* Update datasets.py
3 years ago
0e341c5660
Create one_cycle() function ( #1836 )
3 years ago
d5289b54c4
clean_str() function addition ( #1674 )
* clean_str() function addition
* cleanup
* add euro symbol €
* add closing exclamation (spanish)
* cleanup
3 years ago
1fc9d42a64
NMS --classes 0 bug fix ( #1710 )
3 years ago
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
* Simplify autoshape() post-process
* cleanup
* cleanup
3 years ago
68e6ab668b
Hub device mismatch bug fix ( #1619 )
3 years ago
96a84468b9
Update labels_to_image_weights() ( #1545 )
4 years ago
95fa65339f
Cat apriori to autolabels ( #1484 )
4 years ago
394131c2aa
Use torchvision.ops.nms ( #1460 )
4 years ago
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
4 years ago
548a98a391
Recursive directories dataset capability ( #1408 )
* Recursive directories dataset capability
* x.split('.')[-1]
* f += glob.glob(f"{p}{os.sep}**.*", recursive=True)
* f += glob.glob(str(p / '**.*'), recursive=True)
* remove . from image and vid formats
* .txt to txt
* str(p / '**' / '*.*')
4 years ago
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 years ago
c4addd7761
Unified '/project/name' results saving ( #1377 )
* Project/name update
* Update ci-testing.yml
* address project with path separator failure mode
* Project/name update
* address project with path separator failure mode
* Update ci-testing.yml
* detect.py default --name bug fix
* missing rstrip PR
* train/exp0 to train/exp
4 years ago
67bf9a974e
Replace 'ground truth' with 'labels' ( #1337 )
* Replace 'ground truth' with 'labels'
* Using torch... update
4 years ago
4821d076e2
Increment train, test, detect runs/ ( #1322 )
* Increment train, test, detect runs/
* Update ci-testing.yml
* inference/images to data/images
* move images
* runs/exp to runs/train/exp
* update 'results saved to %s' str
4 years ago
ea7e78cb11
Grid indices overflow bug fix ( #1316 )
4 years ago
fed5d9cdfb
Grid indices overflow bug fix ( #1313 )
4 years ago
453acdec67
Update tensorboard logging
4 years ago
8e9b3d0205
Precision-Recall Curve feature update #1206
4 years ago
ed85038c38
Precision-Recall Curve feature update ( #1206 )
* Precision-Recall Curve feature update
* sentinel value update
4 years ago
c67e72200e
fix compatibility for hyper config ( #1146 )
* fix/hyper
* Hyp giou check to train.py
* restore general.py
* train.py overwrite fix
* restore general.py and pep8 update
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
Glenn Jocher
NanoCode012
Tommy in Tongji
Polydefkis Gkagkos
Jirka Borovec