Glenn Jocher
e80a09bbfa
Update ci-testing.yml to Python 3.9 ( #5660 )
2021-11-15 20:15:50 +01:00
Glenn Jocher and pre-commit-ci[bot]
80cfaf40ef
AutoAnchor and AutoBatch LOGGER ( #5635 )
...
* AutoBatch, AutoAnchor `LOGGER`
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update autoanchor.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-13 15:40:18 +01:00
Glenn Jocher
7473f0f95d
DDP WORLD_SIZE-safe dataloader workers ( #5631 )
...
* WORLD_SIZE-safe workers
* Update with DDP comment
2021-11-12 14:48:55 +01:00
Glenn Jocher
d5b21b1ecb
HUB dataset previews to JPEG ( #5627 )
...
@kalenmike per our convo yesterday.
2021-11-12 12:05:25 +01:00
Glenn Jocher and pre-commit-ci[bot]
def7a0fd19
Improve plots.py robustness ( #5616 )
...
* Improve plots.py robustness
Addresses issues #5374 , #5395 , #5611
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-11 12:56:38 +01:00
Glenn Jocher
69032519bc
np.unique() sort fix for segments (#5609 )
...
* `np.unique()` sort fix for segments
* Update datasets.py
2021-11-11 00:15:17 +01:00
Glenn Jocher
30bc089cbb
Update val.py speed and study tasks ( #5608 )
...
Accepts all arguments now by default resolving https://github.com/ultralytics/yolov5/issues/5600
2021-11-10 16:48:38 +01:00
Glenn Jocher and pre-commit-ci[bot]
61c50199a2
Update train, val tqdm to fixed width ( #5367 )
...
* Update tqdm for fixed width
* Update val.py
* Update val.py
* Try ncols= in train.py
* NCOLS
* NCOLS
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* bar_format
* position 0 leave true
* exp0
* auto
* auto
* Cleanup
* Cleanup
* Cleanup
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-10 15:47:38 +01:00
Glenn Jocher and pre-commit-ci[bot]
7ebb5e5da6
Add notebook_init() to utils/__init__.py ( #5488 )
...
* Update __init__.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* notebook_init
* notebook_init
* notebook_init
* notebook_init
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* notebook_init
* Created using Colaboratory
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-09 23:03:19 +01:00
Glenn Jocher
7207fe95e5
FROM nvcr.io/nvidia/pytorch:21.10-py3 ( #5592 )
2021-11-09 17:55:57 +01:00
Glenn Jocher and pre-commit-ci[bot]
3883261143
New DetectMultiBackend() class ( #5549 )
...
* New `DetectMultiBackend()` class
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* pb to pt fix
* Cleanup
* explicit apply_classifier path
* Cleanup2
* Cleanup3
* Cleanup4
* Cleanup5
* Cleanup6
* val.py MultiBackend inference
* warmup fix
* to device fix
* pt fix
* device fix
* Val cleanup
* COCO128 URL to assets
* half fix
* detect fix
* detect fix 2
* remove half from DetectMultiBackend
* training half handling
* training half handling 2
* training half handling 3
* Cleanup
* Fix CI error
* Add torchscript _extra_files
* Add TorchScript
* Add CoreML
* CoreML cleanup
* New `DetectMultiBackend()` class
* pb to pt fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* explicit apply_classifier path
* Cleanup2
* Cleanup3
* Cleanup4
* Cleanup5
* Cleanup6
* val.py MultiBackend inference
* warmup fix
* to device fix
* pt fix
* device fix
* Val cleanup
* COCO128 URL to assets
* half fix
* detect fix
* detect fix 2
* remove half from DetectMultiBackend
* training half handling
* training half handling 2
* training half handling 3
* Cleanup
* Fix CI error
* Add torchscript _extra_files
* Add TorchScript
* Add CoreML
* CoreML cleanup
* revert default to pt
* Add Usage examples
* Cleanup val
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-09 16:45:02 +01:00
Glenn Jocher and pre-commit-ci[bot]
79bca2bf64
LOGGER consolidation (#5569 )
...
* Logger consolidation
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-08 16:32:15 +01:00
Glenn Jocher
0de4a9c35d
Add --conf-thres >> 0.001 warning ( #5567 )
...
Partially addresses invalid mAPs at higher confidence threshold issue https://github.com/ultralytics/yolov5/issues/1466 .
2021-11-08 16:04:31 +01:00
Glenn Jocher
b8f979bafa
Inside Ultralytics video https://youtu.be/Zgi9g1ksQHc ( #5546 )
...
* Update detect.py Usage examples
* Inside Ultralytics at https://youtu.be/Zgi9g1ksQHc
2021-11-06 20:34:54 +01:00
Glenn Jocher and pre-commit-ci[bot]
3f64ad1760
Fix save_one_box() ( #5545 )
...
* Fix `save_one_box()`
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-06 19:28:03 +01:00
Glenn Jocher
60c8a4f696
Fix for *.yaml emojis on load ( #5543 )
...
Fix for Colab hub error:
```python
import yaml
with open('yolov5s.yaml', errors='ignore') as f:
d = yaml.safe_load(f) # model dict
print(d)
---------------------------------------------------------------------------
ReaderError Traceback (most recent call last)
<ipython-input-8-1150b5143073> in <module>()
2
3 with open('yolov5s.yaml', errors='ignore') as f:
----> 4 d = yaml.safe_load(f) # model dict
5
6 print(d)
6 frames
/usr/local/lib/python3.7/dist-packages/yaml/reader.py in check_printable(self, data)
142 position = self.index+(len(self.buffer)-self.pointer)+match.start()
143 raise ReaderError(self.name, position, ord(character),
--> 144 'unicode', "special characters are not allowed")
145
146 def update(self, length):
ReaderError: unacceptable character #x1f680: special characters are not allowed
in "yolov5s.yaml", position 9
```
2021-11-06 16:03:14 +01:00
Glenn Jocher
e189fa15ea
intersect_dicts() in hubconf.py fix (#5542 )
2021-11-06 15:41:17 +01:00
Glenn Jocher
fa2344cdd8
Update models/hub/*.yaml files for v6.0n release ( #5540 )
...
* Update model yamls for v6.0
* Add python models/yolo.py --test
* Ghost fix
2021-11-06 15:07:45 +01:00
Glenn Jocher
76d90d899a
Update Issue Templates with 💡 ProTip! ( #5539 )
...
* Update bug-report.yml
* Update question.yml
* Update bug-report.yml
2021-11-06 13:58:12 +01:00
Glenn Jocher
cb18cac33d
Update autobatch.py ( #5538 )
...
* Update autobatch.py
* Update autobatch.py
* Update autobatch.py
2021-11-06 13:49:00 +01:00
Glenn Jocher
60e42e16c2
Update autobatch.py ( #5536 )
2021-11-06 12:21:17 +01:00
Glenn Jocher
32b8738735
Update check_file() avoid repeat URL downloads ( #5526 )
2021-11-05 19:22:47 +01:00
Glenn Jocher and pre-commit-ci[bot]
5f603a9dba
Fix detect.py URL inference ( #5525 )
...
* Fix detect.py URL inference
Allows detect.py to run inference on remote URL sources, i.e.:
```python
!python detect.py --weights yolov5s.pt --source https://ultralytics.com/assets/zidane.jpg # image URL
!python detect.py --weights yolov5s.pt --source https://ultralytics.com/assets/decelera_landscape.mov # video URL
```
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2021-11-05 18:43:03 +01:00
Glenn Jocher
853505339a
Fix increment_path() explicit file vs dir handling ( #5523 )
...
Resolves https://github.com/ultralytics/yolov5/pull/5341#issuecomment-961774729
Tests:
```python
import glob
import re
from pathlib import Path
def increment_path(path, exist_ok=False, sep='', mkdir=False):
# Increment file or directory path, i.e. runs/exp --> runs/exp{sep}2, runs/exp{sep}3, ... etc.
path = Path(path) # os-agnostic
if path.exists() and not exist_ok:
path, suffix = (path.with_suffix(''), path.suffix) if path.is_file() else (path, '')
dirs = glob.glob(f"{path}{sep}*") # similar paths
matches = [re.search(rf"%s{sep}(\d+)" % path.stem, d) for d in dirs]
i = [int(m.groups()[0]) for m in matches if m] # indices
n = max(i) + 1 if i else 2 # increment number
path = Path(f"{path}{sep}{n}{suffix}") # increment path
if mkdir:
path.mkdir(parents=True, exist_ok=True) # make directory
return path
print(increment_path('runs'))
print(increment_path('export.py'))
print(increment_path('abc.def.dir'))
print(increment_path('abc.def.file'))
```
2021-11-05 15:46:20 +01:00
Glenn Jocher
bcc085d83f
Common is_coco logic betwen train.py and val.py ( #5521 )
2021-11-05 13:32:46 +01:00
Glenn Jocher
bfacfc6b4a
Update cls bias init ( #5520 )
...
* Update cls bias init
Increased numerical precision. Returns 1.0 probability for single-class datasets now. Addresses https://github.com/ultralytics/yolov5/issues/5357
```python
torch.sigmoid(torch.tensor([math.log(0.6 / (1 - 0.99999))]))
Out[19]: tensor([1.0000])
```
* Update yolo.py
2021-11-05 13:18:46 +01:00
Glenn Jocher
17b5f5b974
Fix increment_path() with periods ( #5518 )
...
Fix for https://github.com/ultralytics/yolov5/issues/5503
```
python detect.py --visualize --source path/to/file.suffix1.jpg
```
2021-11-05 11:16:19 +01:00
Glenn Jocher
5866646cc8
Fix float zeros format ( #5491 )
...
* Fix float zeros format
* 255 to integer
2021-11-03 23:36:53 +01:00
Glenn Jocher
8a803f36d3
Delete code-format.yml ( #5487 )
2021-11-03 21:06:32 +01:00
Glenn Jocher and pre-commit
34b859a41e
Keras CI fix ( #5486 )
...
* Keras CI fix
* pre-commit fixes
* Update ci-testing.yml
Co-authored-by: pre-commit <pre-commit@example.com >
2021-11-03 19:25:44 +01:00
Glenn Jocher
06bf8ef7e5
Add tf.py verification printout ( #5484 )
...
* Update tf.py with verified confirmation
* Update ci-testing.yml
* Update ci-testing.yml
2021-11-03 19:04:51 +01:00
Glenn Jocher
62d77a1027
Created using Colaboratory
2021-11-03 18:55:25 +01:00
Glenn Jocher
84a8099b75
Update torch_utils.py import LOGGER ( #5483 )
2021-11-03 17:17:38 +01:00
Glenn Jocher
df30426c03
Improve GPU name ( #5478 )
...
* Improve GPU name
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
2021-11-03 16:59:53 +01:00
Glenn Jocher
0eb37ad8af
Remove check_requirements(('tensorflow>=2.4.1',)) ( #5476 )
...
`check_requirements()` is unreliable for large packages like torch and tensorflow that may have multiple installation routes (i.e. conda, pip, tensorflow-cpu, etc.)
2021-11-03 14:51:37 +01:00
Glenn Jocher
042f02ff9b
Fix tf.py LoadImages() dataloader return values ( #5455 )
2021-11-02 23:04:15 +01:00
Glenn Jocher
4c0982a243
Update README.md ( #5438 )
...
2-line update
2021-11-01 18:28:14 +01:00
Glenn Jocher and pre-commit
7b1f7aec46
Update get_loggers() ( #4854 )
...
* Update `set_logging()`
* Update export.py
* pre-commit fixes
* Update LoadImages
* Update LoadStreams
* Update print_args
* Single LOGGER definition
* yolo.py fix
Co-authored-by: pre-commit <pre-commit@example.com >
2021-11-01 18:22:13 +01:00
Glenn Jocher
8c326a1edf
Meshgrid indexing='ij' for PyTorch 1.10 ( #5309 )
...
* Meshgrid `indexing='ij'` for PyTorch 1.10
Will not merge currently as breaks backwards compatibility.
* Meshgrid `indexing='ij'` for PyTorch 1.10
Will not merge currently as breaks backwards compatibility.
* Add check_version hard argument
* Update comment
2021-11-01 14:33:08 +01:00
Glenn Jocher
5d4258fac5
Fix MixConv2d() remove shortcut + apply depthwise ( #5410 )
2021-10-30 13:38:51 +02:00
Glenn Jocher and pre-commit
7f9bbf0268
Update GitHub issues templates ( #5404 )
...
* Update GitHub issues templates
* pre-commit fixes
Co-authored-by: pre-commit <pre-commit@example.com >
2021-10-29 23:16:04 +02:00
Glenn Jocher
a4fece8c14
DDP nl fix ( #5332 )
2021-10-25 16:03:39 +02:00
Glenn Jocher
9c31a66f5d
Update AutoShape.forward() model.classes example ( #5324 )
2021-10-25 14:03:25 +02:00
Glenn Jocher
ca19df5f7f
Add autobatch feature for best batch-size estimation ( #5092 )
...
* Autobatch
* fix mem
* fix mem2
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update train.py
* print result
* Cleanup print result
* swap fix in call
* to 64
* use total
* fix
* fix
* fix
* fix
* fix
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Cleanup printing
* Update final printout
* Update autobatch.py
* Update autobatch.py
* Update autobatch.py
2021-10-25 13:56:13 +02:00
Glenn Jocher
79d8f1f678
Created using Colaboratory
2021-10-22 21:19:23 +02:00
Glenn Jocher
b760acec11
Created using Colaboratory
2021-10-22 21:11:29 +02:00
Glenn Jocher
441b47c443
More informative EarlyStopping() message ( #5303 )
2021-10-22 20:02:19 +02:00
Glenn Jocher
a18b0c36cd
Update rebase.yml with workflows permissions ( #5255 )
...
Update token with workflows permissions
2021-10-19 12:02:15 +02:00
Glenn Jocher
4f5e274eee
Update rebase.yml ( #5245 )
2021-10-19 09:46:05 +02:00
Glenn Jocher
752d97df2c
Add on_fit_epoch_end callback ( #5232 )
...
* Add `on_fit_epoch_end` callback
* Add results to train
* Update __init__.py
2021-10-18 14:28:22 +02:00
Glenn Jocher
13f7275555
Update/inplace ops ( #5233 )
...
* Clip Objects365 autodownload labels (#5214 )
Fixes out of bounds labels that seem to affect ~10% of images in dataset.
* Inplace ops
2021-10-18 14:24:48 +02:00
Glenn Jocher
0000334a6f
Update Objects365.yaml val count ( #5212 )
2021-10-15 12:40:40 -07:00
Glenn Jocher
991c654e81
Autofix duplicate label handling ( #5210 )
...
* Autofix duplicate labels
PR changes duplicate label handling from report error and ignore image-label pair to report warning and autofix image-label pair.
This should fix this common issue for users and allow everyone to get started and get a model trained faster and easier than before.
* sign fix
* Cleanup
* Increment cache version
* all to any fix
2021-10-15 12:32:08 -07:00
Glenn Jocher
1c9f710b46
Add hyp.scratch-med.yaml ( #5196 )
...
* Add hyp.scratch-med.yaml
* Update hyp.scratch-med.yaml
2021-10-14 20:40:30 -07:00
Glenn Jocher
6b31cfd793
Highlight contributors in README ( #5173 )
...
* Highlight contributors in README
* Adjust Contact section
* Update README.md
2021-10-14 12:49:16 -07:00
Glenn Jocher
7ee5aed0b3
Improved check_suffix() robustness to '' and "" ( #5192 )
...
* Improved check_suffix() robustness to `''` and `""`
* Cleanup
2021-10-14 12:00:39 -07:00
Glenn Jocher
e520e820f1
Update autodownload fallbacks to v6.0 assets ( #5177 )
2021-10-13 19:58:38 -07:00
Glenn Jocher
a346926996
Add class filtering to LoadImagesAndLabels() dataloader ( #5172 )
...
* Add train class filter feature to datasets.py
Allows for training on a subset of total classes if `include_class` list is defined on datasets.py L448:
```python
include_class = [] # filter labels to include only these classes (optional)
```
* segments fix
2021-10-13 15:48:54 -07:00
Glenn Jocher
b754525e99
Check 'onnxruntime-gpu' if torch.has_cuda ( #5087 )
...
* Check `'onnxruntime-gpu' if torch.has_cuda`
* fix indent
2021-10-12 22:25:38 -07:00
Glenn Jocher
20a809dee5
Update stale.yml ( #5156 )
2021-10-12 17:54:56 -07:00
Glenn Jocher
26784afd99
Use YOLOv5n for CI testing ( #5154 )
2021-10-12 08:42:26 -07:00
Glenn Jocher
956be8e642
YOLOv5 release v6.0 ( #5141 )
...
* Update P5 models
* Update P6 models
* Update with GFLOPs and Params
* Update with GFLOPs and Params
* Update README
* Update
* Update README
* Update
* Update
* Add times
* Update README
* Update results
* Update results
* Update results
* Update hyps
* Update plots
* Update plots
* Update README.md
* Add nano models to hubconf.py
2021-10-11 23:47:18 -07:00
Glenn Jocher
0bf24cf641
Add OpenCV DNN option for ONNX inference ( #5136 )
...
* Add OpenCV DNN option for ONNX inference
Usage:
```bash
python detect.py --weights yolov5s.onnx # ONNX Runtime inference
python detect.py --weights yolov5s.onnx -dnn # OpenCV DNN inference
```
* DNN prediction to tensor
* Update detect.py
2021-10-11 12:39:20 -07:00
Glenn Jocher
4cf7d487a6
Update plots.py ( #5127 )
2021-10-11 01:59:22 -07:00
Glenn Jocher
7d37b3c52e
Update val.py pad = 0.0 if task == speed else 0.5 ( #5121 )
...
* Update val.py `pad = 0.0 if task == speed else 0.5`
* Cleanup
2021-10-10 23:20:42 -07:00
Glenn Jocher
ba4b79de8b
Update val.py --speed and --study usages ( #5120 )
2021-10-10 21:15:28 -07:00
Glenn Jocher
4f9718abe6
Pass --device for --task study ( #5118 )
2021-10-10 15:07:26 -07:00
Glenn Jocher
276b674513
Fix SKU-110K HUB: OSError ( #5106 )
2021-10-09 00:12:59 -07:00
Glenn Jocher
57ce73cd4f
Update datasets.py comments ( #5088 )
2021-10-07 16:04:08 -07:00
Glenn Jocher
ce8e5dc8bb
Update Dockerfile to ADD Arial.ttf ( #5084 )
...
* Update Dockerfile to `ADD` Arial.ttf
Fix for `Arial.ttf download in Docker` #5078
* Cleanup
* Cleanup2
2021-10-07 14:16:33 -07:00
Glenn Jocher
070af88108
Fix yaml.safe_load() ignore emoji errors ( #5060 )
2021-10-05 13:41:52 -07:00
Glenn Jocher
5afc9c25ef
Implement --save-period locally ( #5047 )
...
This PR adds a new training argument `--save-period` to save training checkpoints every `x` epochs. To save training every 50 epochs for example:
```
python train.py --save-period 50 # saves epoch50.pt, epoch100.pt, epoch150.pt, ... etc.
```
This saved checkpoints in addition to existing last.pt and best.pt checkpoints and does not affect their behavior. Default value is -1, i.e. disabled.
2021-10-04 18:48:46 -07:00
Glenn Jocher
d133968b1e
Clip TTA Augmented Tails ( #5028 )
...
* Clip TTA Augmented Tails
Experimental TTA update.
* Update yolo.py
* Update yolo.py
* Update yolo.py
* Update yolo.py
2021-10-04 15:48:55 -07:00
Glenn Jocher
b0ade48457
Fix missing opt.device on --task study ( #5031 )
2021-10-02 17:55:55 -07:00
Glenn Jocher
cce7e78fae
Created using Colaboratory
2021-09-30 19:10:55 -07:00
Glenn Jocher
89a6e9ffa7
Compute loss on final val ( #5017 )
2021-09-30 15:16:11 -07:00
Glenn Jocher
b20e381492
Update README.md ( #5015 )
2021-09-30 14:06:00 -07:00
Glenn Jocher
452e84703c
Update train.py ( #5014 )
2021-09-30 13:57:53 -07:00
Glenn Jocher
ba99092304
Update W&B README ( #5006 )
2021-09-30 00:35:00 -07:00
Glenn Jocher
1916226841
Created using Colaboratory
2021-09-29 22:18:16 -07:00
Glenn Jocher
42354d7025
Update README.md
2021-09-29 16:56:10 -07:00
Glenn Jocher
cd35a009ba
Revert torch.hub.load() test ( #4986 )
...
Temporarily reverts https://github.com/ultralytics/yolov5/pull/4978 until torch 1.10 is released, which should resolve `urllib.error.HTTPError: HTTP Error 403: rate limit exceeded` errors generated by torch hub from GitHub actions runners.
2021-09-28 20:11:20 -07:00
Glenn Jocher
9988059b10
Clickable CI badge ( #4985 )
2021-09-28 17:46:14 -07:00
Glenn Jocher
6b19f728a1
Created using Colaboratory
2021-09-28 17:33:48 -07:00
Glenn Jocher
cff7d2af45
Created using Colaboratory
2021-09-28 17:29:05 -07:00
Glenn Jocher
0c93ec762a
Faster --img 64 CI tests ( #4979 )
2021-09-28 12:41:28 -07:00
Glenn Jocher
94705a9528
Add Hub custom models to CI tests ( #4978 )
...
* Update ci-testing.yml for Hub custom model tests
* Update ci-testing.yml
2021-09-28 12:28:18 -07:00
Glenn Jocher
38c779b099
Created using Colaboratory
2021-09-28 10:04:56 -07:00
Glenn Jocher
fb982d6030
Fix relative ROOT Pytorch Hub custom model bug ( #4974 )
...
* Fix relative `ROOT` Pytorch Hub custom model bug
* Update yolo.py
2021-09-28 06:36:12 -07:00
Glenn Jocher
2993c3fa7a
Add roboflow ( #4956 )
2021-09-27 19:00:05 -07:00
Glenn Jocher
3aeb57d66f
Created using Colaboratory
2021-09-27 18:32:22 -07:00
Glenn Jocher
5ed28603cf
Update relative ROOT logic ( #4955 )
...
* Update relative `ROOT` logic
* python 3.9 Path().is_relative_to() removal
2021-09-27 18:26:41 -07:00
Glenn Jocher
4e04cb0dc8
Allow YOLOv5 execution from arbitrary cwd ( #4954 )
...
* Allow YOLOv5 execution from arbitrary `cwd`
* Fix str bugs
2021-09-27 17:40:20 -07:00
Glenn Jocher
a820b43aca
Automatic Chinese fonts plotting ( #4951 )
...
* Automatic Chinese fonts plotting
* Default PIL=False
2021-09-27 13:48:15 -07:00
Glenn Jocher
c5ba2abb4a
Update Dockerfile ( #4935 )
2021-09-26 19:16:14 -07:00
Glenn Jocher
5a8e4343d8
Scope check_file() search space ( #4933 )
...
`check_file()` is now limited to searching opt-in directories: /data, /models, /utils. This prevents large non-project directories like /.git and /venv from being searched, which may cause `check_file()` to slow significantly.
2021-09-26 17:11:46 -07:00
Glenn Jocher
793383232f
Update default Albumentations ( #4931 )
2021-09-26 12:10:33 -07:00
Glenn Jocher
d856c48298
Validate best.pt on train end ( #4889 )
...
* Validate best.pt on train end
* 0.7 iou for COCO only
* pass callbacks
* active model.float() if not half
* print Validating best.pt...
* add newline
2021-09-26 11:54:53 -07:00
Glenn Jocher
e687873436
Add Slack Forum badge to README ( #4930 )
...
Add badge with link to join the new YOLOv5 Slack Forum!
https://join.slack.com/t/ultralytics/shared_invite/zt-w29ei8bp-jczz7QYUmDtgo6r6KcMIAg
2021-09-26 10:54:55 -07:00
Glenn Jocher
39c17ce0b9
Fix root referenced before assignment ( #4920 )
...
* Fix `root` referenced before assignment
Fix for bug introduced by #4919 discovered on VOC autodownload:
```
python train.py --data VOC.yaml
```
* Cleanup
2021-09-25 09:16:14 -07:00