317f2ccc9d
Improved `check_requirements()` offline-handling ( #3466 )
Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
3年前
563ea9475a
Add `check_git_status()` 5 second timeout ( #3464 )
* Add check_git_status() 5 second timeout
This should prevent the SSH Git bug that we were discussing @KalenMike
* cleanup
* replace timeout with check_output built-in timeout
3年前
d40481acc5
Add `timeout()` class ( #3460 )
* Add `timeout()` class
* rearrange order
3年前
f8651c388f
Parse URL authentication ( #3424 )
* Parse URL authentication
* urllib.parse.unquote()
* improved error handling
* improved error handling
* remove %3F
* update check_file()
3年前
d833ab3d25
Create `is_pip()` function ( #3391 )
Returns `True` if file is part of pip package. Useful for contextual behavior modification.
```python
def is_pip():
# Is file in a pip package?
return 'site-packages' in Path(__file__).absolute().parts
```
3年前
2435bfe896
Add URL download to check_file() ( #3330 )
* Add URL file download to check_file()
* cleanup
* pathlib bug fix
3年前
9f3a388cea
Improved check_requirements() robustness ( #3298 )
Add try: except clause on missing requirements install to catch install failures. Prompted by Ultralytics YOLOv5 API failure on Cython.
3年前
a039b7796d
PyTorch Hub `crops = results.crop()` return values ( #3282 )
* Changing save_one_box
Made to work with other changes to common.py
* PEP8 and single line BGR
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3年前
3f74cd9ed1
Parameterize max_det + inference default at 1000 ( #3215 )
* Added max_det parameters in various places
* 120 character line
* PEP8
* 120 character line
* Update inference default to 1000 instances
* Update inference default to 1000 instances
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3年前
57b0d3a605
Add check_python() ( #3088 )
* Add check_python()
Checks python version against minimum version of 3.7.0.
* remove packaging dependency
* refactor import
3年前
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
3年前
5189b3addb
Objects365 Update
3年前
41cc7caee6
Add NMS threshold checks ( #3020 )
* Add NMS threshold checks
* fix
3年前
fcd57028d2
Add is_colab() function ( #3018 )
3年前
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
3年前
4d7bca7648
Add unzip flag to download() ( #3002 )
3年前
54652fe3ff
Objects365 update
3年前
1849916726
Check_requirements() enclosing apostrophe bug fix ( #2929 )
This fixes a bug where the '>' symbol in python package requirements was not running correctly with subprocess.check_output() commands.
3年前
c0d3f80544
Add verbose option to pytorch hub models ( #2926 )
* Add verbose and update print to logging
* Fix positonal param
* Revert auto formatting changes
* Update hubconf.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3年前
45632b2704
Update download() for tar.gz files ( #2919 )
* Update download() for tar.gz files
* Update general.py
3年前
1b1ab4cca2
Add file_size() function ( #2911 )
* Add file_size() function
* Update export.py
3年前
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
3年前
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3年前
c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub ( #2827 )
* Update detect.py
* Update detect.py
* Update greetings.yml
* Update cropping
* cleanup
* Update increment_path()
* Update common.py
* Update detect.py
* Update detect.py
* Update detect.py
* Update common.py
* cleanup
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3年前
c5c647e281
Update increment_path() to handle file paths ( #2867 )
3年前
1f3e482bce
ONNX Simplifier ( #2815 )
* ONNX Simplifier
Add ONNX Simplifier to ONNX export pipeline in export.py. Will auto-install onnx-simplifier if onnx is installed but onnx-simplifier is not.
* Update general.py
3年前
0f395b3e3b
YOLOv5 v5.0 Release patch 1 ( #2764 )
* torch.jit.trace(model, img, strict=False)
* Update check_file()
* Update hubconf.py
* Update README.md
3年前
e2b7bc0b32
YouTube Livestream Detection ( #2752 )
* Youtube livestream detection
* dependancy update to auto install pafy
* Remove print
* include youtube_dl in deps
* PEP8 reformat
* youtube url check fix
* reduce lines
* add comment
* update check_requirements
* stream framerate fix
* Update README.md
* cleanup
* PEP8
* remove cap.retrieve() failure code
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3年前
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年前
d4456e43b2
Update segment2box() comment ( #2600 )
3年前
333ccc5b0f
YOLOv5 PyTorch Hub models >> check_requirements() ( #2592 )
Improved user-feedback following requirements auto-update.
3年前
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年前
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年前
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年前
ba18528b47
bbox_iou() stability and speed improvements ( #2385 )
3年前
fab5085674
EMA bug fix 2 ( #2330 )
* EMA bug fix 2
* update
3年前
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年前
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年前
d2e754b67b
Add isdocker() ( #2232 )
* Add isdocker()
* Update general.py
* Update general.py
3年前
26c2e54c8f
Add check_imshow() ( #2231 )
* Add check_imshow()
* Update general.py
* Update general.py
3年前
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年前
a5d5f9262d
Changed socket port and added timeout ( #2176 )
3年前
f59f80114c
Add 'exclude' tuple to check_requirements() ( #2041 )
3年前
3b7feeafdf
Update general.py check_git_status() fix ( #2020 )
3年前
aac33f87ad
Update inference multiple-counting ( #2019 )
* Update inference multiple-counting
* update github check
3年前
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年前
d9212140b3
Add xywhn2xyxy() ( #1983 )
3年前
b1cf25dd9a
check_git_status() asserts ( #1977 )
3年前
b26a2f6242
check_git_status() when not exist /workspace ( #1966 )
3年前
e8a41e8916
prevent check_git_status() in docker images ( #1951 )
* prevent check_git_status() running docker images
* Update general.py
3年前
Glenn Jocher
yeric1789
Adrian Holovaty
NanoCode012
Burhan
Ben Milanko