Glenn Jocher
d5289b54c4
clean_str() function addition ( #1674 )
* clean_str() function addition
* cleanup
* add euro symbol €
* add closing exclamation (spanish)
* cleanup
3 år sedan
Polydefkis Gkagkos
1fc9d42a64
NMS --classes 0 bug fix ( #1710 )
3 år sedan
Glenn Jocher
fa8f1fb0e9
Simplify autoshape() post-process ( #1653 )
* Simplify autoshape() post-process
* cleanup
* cleanup
3 år sedan
Glenn Jocher
68e6ab668b
Hub device mismatch bug fix ( #1619 )
3 år sedan
Glenn Jocher
96a84468b9
Update labels_to_image_weights() ( #1545 )
4 år sedan
Glenn Jocher
95fa65339f
Cat apriori to autolabels ( #1484 )
4 år sedan
Glenn Jocher
394131c2aa
Use torchvision.ops.nms ( #1460 )
4 år sedan
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
4 år sedan
Glenn Jocher
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 år sedan
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 år sedan
Glenn Jocher
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 år sedan
Glenn Jocher
67bf9a974e
Replace 'ground truth' with 'labels' ( #1337 )
* Replace 'ground truth' with 'labels'
* Using torch... update
4 år sedan
Glenn Jocher
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 år sedan
Glenn Jocher
ea7e78cb11
Grid indices overflow bug fix ( #1316 )
4 år sedan
Glenn Jocher
fed5d9cdfb
Grid indices overflow bug fix ( #1313 )
4 år sedan
Glenn Jocher
453acdec67
Update tensorboard logging
4 år sedan
Glenn Jocher
8e9b3d0205
Precision-Recall Curve feature update #1206
4 år sedan
Glenn Jocher
ed85038c38
Precision-Recall Curve feature update ( #1206 )
* Precision-Recall Curve feature update
* sentinel value update
4 år sedan
Jirka Borovec
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 år sedan
Glenn Jocher
4d3680c81d
Minor import and spelling updates ( #1133 )
4 år sedan
Jirka Borovec
00917a6225
update expt name comment and folder parsing for training ( #978 )
* comment
* fix parsing
* fix evolve
* folder
* tqdm
* Update train.py
* Update train.py
* reinstate anchors into meta dict
anchor evolution is working correctly now
* reinstate logger
prefer the single line readout for concise logging, which helps simplify notebook and tutorials etc.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 år sedan
Glenn Jocher
0ada058f63
Generalized regression criterion renaming ( #1120 )
4 år sedan
Glenn Jocher
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
4 år sedan
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
4 år sedan
Jiacong Fang
c5d2331897
Fix increment_dir to use run_xxx for logdir ( #1058 )
* Fix increment_dir to use run_xxx for logdir
Rerunning train.py with logdir in the form of `run_xxx` causes index slicing in
```
n = max([int(x[len(dir):x.find('_') if '_' in Path(x).name else None]) for x in d]) + 1 # increment
```
to underflow.
* Replace find with rfind
4 år sedan
Glenn Jocher
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
4 år sedan
Glenn Jocher
c54e39464c
check_file() assert unique
4 år sedan
Glenn Jocher
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
4 år sedan
Glenn Jocher
ffe9eb4238
remove if platform.system() == 'Darwin' checks
4 år sedan
Glenn Jocher
987c226849
seaborn plotting correlogram addition
4 år sedan
rafale77
9776e70988
torch.ops.torchvision.nms ( #860 )
Don't load the entire torchvision library just for nms when the function is already in the torch library.
4 år sedan
Glenn Jocher
31d6c389f2
Update general.py plot_evolution()
4 år sedan
Willie Maddox
455f7b8f76
Update general.py ( #823 )
Fixes #822
`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
4 år sedan
Glenn Jocher
5a7d79fbe6
Update general.py bbox_iou() with eps improvements ( #736 )
eps reduced to 1e-9 for additional precaution.
4 år sedan
Glenn Jocher
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
4 år sedan
Glenn Jocher
9dbaa1bbbe
Update plot_results()
4 år sedan
Glenn Jocher
ea5e97fb9b
FP16 NMS ( #814 )
4 år sedan
Glenn Jocher
b42e8a531b
optimize imports
4 år sedan
Glenn Jocher
5e0b90de8f
CIoU nan bug fix ( #736 )
4 år sedan
Glenn Jocher
ebafd1ead5
single command --resume ( #756 )
* single command --resume
* else check files, remove TODO
* argparse.Namespace()
* tensorboard lr
* bug fix in get_latest_run()
4 år sedan
Glenn Jocher
916d4aad9a
v3.0 Release ( #725 )
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
4 år sedan
Hatovix
56c2c344ff
Fix list paths ( #721 )
* Add list paths on check_dataset
* missing raise statement
* Update general.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 år sedan
Glenn Jocher
9f482cbcb8
utils.general comment updates/bug fixes
4 år sedan
NanoCode012
4949401a94
Fix redundant outputs via Logging in DDP training ( #500 )
* Change print to logging
* Clean function set_logging
* Add line spacing
* Change leftover prints to log
* Fix scanning labels output
* Fix rank naming
* Change leftover print to logging
* Reorganized DDP variables
* Fix type error
* Make quotes consistent
* Fix spelling
* Clean function call
* Add line spacing
* Update datasets.py
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 år sedan
Glenn Jocher
41523e2c91
Dataset autodownload feature addition ( #685 )
* initial commit
* move download scripts into data/scripts
* new check_dataset() function in general.py
* move check_dataset() out of with context
* Update general.py
* DDP update
* Update general.py
4 år sedan
Glenn Jocher
3c6e2f7668
Single-source training ( #680 )
* Single-source training
* Extract hyperparameters into seperate files
* weight decay scientific notation yaml reader bug fix
* remove import glob
* intersect_dicts() implementation
* 'or' bug fix
* .to(device) bug fix
4 år sedan
Glenn Jocher
7eaf225d55
zero-target training bug fix ( #609 )
4 år sedan
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
4 år sedan
Glenn Jocher
ec7a926163
AutoAnchor update to display anchors/target
4 år sedan
Glenn Jocher
d989bc9260
remove NBSP
4 år sedan