96fcde40b8
W&B feature improvements ( #1258 )
...
* W&B feature improvements
This PR add:
* Class to id labels. Now, the caption of bounding boxes will display the class name and the class confidence score.
* The project name is set to "Yolov5" and the run name will be set to opt.logdir
* cleanup
* remove parenthesis on caption
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-01 20:54:29 +01:00
ca290dca24
Weights & Biases (W&B) Feature Addition ( #1235 )
...
* Add wandb metric logging and bounding box debugging
* Improve formatting, readability
* Remove mutliple path for init, improve formatting
* Add wandb params
* Remove typecasting in bbox coordinates and reformat
* Cleanup
* add wandb to requirements.txt
* minor updates to test.py
* general reorg
* reduce --log-imgs to 10
* clean wandb import
* reverse wandb import assert
* add except AssertionError to try import
* move wandb init to all global ranks
* replace print() with logger.info()
* replace print() with logger.info()
* move wandb.init() bug fix
* project PosixPath to basename bug fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-31 20:16:35 +01:00
453acdec67
Update tensorboard logging
2020-10-26 01:08:33 +01:00
d3dad42256
Move generated *.json annotations into runs/test ( #1209 )
...
* Move generated *.json annotations into runs/test
* list and string robust
2020-10-25 18:19:44 +01:00
19c8b2c9b9
Adding --save-dir and --save-conf options to test.py ( #1182 )
...
* Adding --output and --save-conf options to test.py
* Update help fields
* Update test.py
* Make arguments and comments uniform with test.py
* Remove previous and print save_dir on finish
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-25 16:50:21 +01: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
c171e45cda
Update test.py with set() ( #806 )
2020-08-21 15:03:03 -07:00
da35727c68
--save-txt bug fix ( #763 )
2020-08-17 17:14:07 -07:00
f5da528d28
reformat code
2020-08-14 11:53:44 -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
0892c44bc4
Fix Logging ( #719 )
...
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-12 14:18:19 -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
d5b6416c87
Explicit Imports ( #498 )
...
* expand imports
* optimize
* miss
* fix
2020-08-02 15:47:36 -07:00
43a616a955
PyTorch 1.6.0 compatability updates
2020-07-28 15:38:56 -07:00
07a82f4d44
Update ci-testing.yml ( #445 )
...
* Update ci-testing.yml
* Update ci-testing.yml
* Update requirements.txt
* Update requirements.txt
* Update google_utils.py
* Update test.py
* Update ci-testing.yml
2020-07-19 15:44:11 -07:00
b40852d5a5
update test.py
2020-07-18 11:31:22 -07:00
37acbdc0b6
update test.py --save-txt
2020-07-15 20:03:41 -07:00
b8c2da4a0d
update test.py --save-txt
2020-07-15 20:00:48 -07:00
1ce686edfc
update test.py
2020-07-15 19:06:39 -07:00
6bd9218922
update test.py
2020-07-15 19:04:15 -07:00
aa08b2bd6f
test.py --task test updates
2020-07-15 18:25:34 -07:00
a1c8406af3
EMA and non_blocking=True
2020-07-13 20:19:10 -07:00
0afbb8d498
update test.py model.half()
2020-07-12 13:25:58 -07:00
bf6f41567a
hyperparameter printout update
2020-07-08 17:21:00 -07:00
dc5e18390a
Merge branch 'master' into advanced_logging
2020-07-08 17:01:19 -07:00
22ab1c295f
Update test.py
2020-07-08 16:29:31 -07:00
c3d3e6b776
Update test.py
2020-07-08 16:27:52 -07:00
e8cf24b6c8
Initial model ensemble capability #318
2020-07-07 15:40:50 -07:00
5ac517b22a
Remove duplicate `verbose` arg in test.py
2020-07-07 10:09:53 -05:00
f517ba81c7
Merge branch 'master' into advanced_logging
2020-07-06 16:52:11 -05:00
6b95d6d2c0
.to(device) bug fix
2020-07-05 23:37:36 -07:00
a62333e808
test.py .fuse() update
2020-07-05 23:27:03 -07:00
04bdbe4104
fuse update
2020-07-05 23:16:50 -07:00
a9dc0c2c29
update check_img_size() for model strides
2020-06-30 15:25:38 -07:00
e18e6811dc
Merge branch 'master' into advanced_logging
2020-06-27 10:13:03 -05:00
22fb2b0c25
refactor dataloader
2020-06-26 18:56:13 -07:00
e670a3353b
update test.py for FP16 testing during training
2020-06-25 15:09:04 -07:00
227aa735dc
updates
2020-06-24 13:20:14 -07:00
b8557f87e3
add stride to datasets.py
2020-06-24 13:02:27 -07:00
a63e1c9b3c
study update
2020-06-22 23:23:50 -07:00
db898131f9
study update
2020-06-22 23:07:22 -07:00
6c1b87a42e
update google_utils import
2020-06-22 23:00:23 -07:00
1f1917ef56
remove fast, add merge
2020-06-21 13:37:11 -07:00
9b7386f603
Add save_dir arg to test.test, use arg as location for saving batch jpgs
2020-06-17 16:08:46 -05:00
2daa412575
train with multi-gpu half test bug fix #99
2020-06-16 16:28:25 -07:00
d1e5716610
train with multi-gpu half test bug fix #99
2020-06-16 16:22:58 -07:00
72d06147b2
Multi-GPU disabled in test.py
2020-06-16 14:42:30 -07:00
7c6fd4d670
--verbose bug fix #84
2020-06-16 10:22:22 -07:00
db2c3acd3a
updated testing settings, rebalanced towards FP16 latency
2020-06-15 16:55:57 -07:00
Ayush Chaurasia
Glenn Jocher
oleg
NanoCode012
Jirka Borovec
Alex Stoken