ed85038c38
Precision-Recall Curve feature update ( #1206 )
...
* Precision-Recall Curve feature update
* sentinel value update
2020-10-25 12:55:52 +01:00
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>
2020-10-15 15:05:58 +02:00
4d3680c81d
Minor import and spelling updates ( #1133 )
2020-10-13 17:24:27 +02:00
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>
2020-10-13 14:10:21 +02:00
0ada058f63
Generalized regression criterion renaming ( #1120 )
2020-10-11 17:25:17 +02:00
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
...
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
2020-10-09 14:50:59 +02:00
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
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
2020-10-05 15:46:32 +02:00
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
2020-09-13 14:08:06 -07:00
c54e39464c
check_file() assert unique
2020-09-11 14:33:06 -07:00
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
2020-09-03 20:29:08 -07:00
ffe9eb4238
remove if platform.system() == 'Darwin' checks
2020-09-03 11:41:33 -07:00
987c226849
seaborn plotting correlogram addition
2020-08-29 15:13:58 -07:00
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.
2020-08-27 11:13:41 -07:00
31d6c389f2
Update general.py plot_evolution()
2020-08-25 23:08:14 -07:00
455f7b8f76
Update general.py ( #823 )
...
Fixes #822
`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
2020-08-25 13:01:17 -07:00
5a7d79fbe6
Update general.py bbox_iou() with eps improvements ( #736 )
...
eps reduced to 1e-9 for additional precaution.
2020-08-25 12:56:39 -07:00
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
2020-08-22 15:47:39 -07:00
9dbaa1bbbe
Update plot_results()
2020-08-22 15:32:05 -07:00
ea5e97fb9b
FP16 NMS ( #814 )
2020-08-21 14:44:25 -07:00
b42e8a531b
optimize imports
2020-08-20 21:17:40 -07:00
5e0b90de8f
CIoU nan bug fix ( #736 )
2020-08-20 10:37:55 -07:00
ebafd1ead5
single command --resume ( #756 )
...
* single command --resume
* else check files, remove TODO
* argparse.Namespace()
* tensorboard lr
* bug fix in get_latest_run()
2020-08-17 16:28:43 -07:00
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
2020-08-13 14:25:05 -07:00
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>
2020-08-12 18:53:38 -07:00
9f482cbcb8
utils.general comment updates/bug fixes
2020-08-11 20:11:44 -07:00
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>
2020-08-11 11:18:45 -07:00
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
2020-08-09 20:52:57 -07:00
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
2020-08-09 02:12:44 -07:00
7eaf225d55
zero-target training bug fix ( #609 )
2020-08-05 13:35:31 -07:00
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher
Jirka Borovec
Jiacong Fang
rafale77
Willie Maddox
Hatovix
NanoCode012