Commit Graph

808 Commits (49abc722fc4204d57e81b2f4db98f024ad6474cc)
 

Author SHA1 Message Date
Glenn Jocher ce3872f86c
W&B evolution bug fix (#1373) 2020-11-12 12:55:52 +01:00
Glenn Jocher f6b3c96966
Update tutorial.ipynb 2020-11-11 16:15:20 +01:00
Glenn Jocher 9cd46427e4 Creado con Colaboratory 2020-11-11 16:11:49 +01:00
Glenn Jocher 104f5418ea Creado con Colaboratory 2020-11-10 17:46:36 +01:00
Glenn Jocher 7841d7ba88
Update README.md 2020-11-10 17:12:35 +01:00
Glenn Jocher 6ff0d916fb
Update download_weights.sh (#1338) 2020-11-10 15:04:14 +01:00
Glenn Jocher 27138c2ebb
Update download_weights.sh (#1338) 2020-11-10 14:28:45 +01:00
Glenn Jocher 67bf9a974e
Replace 'ground truth' with 'labels' (#1337)
* Replace 'ground truth' with 'labels'

* Using torch... update
2020-11-10 14:15:03 +01:00
Khiem Doan 1c8464e199
Use pathlib instead of low-level module (#1329)
* Use pathlib instead of low-level module

* Use pathlib instead of low-level module

* Update detect.py

* Update test.py

* reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-10 13:01:19 +01:00
Glenn Jocher 19e2482458
PyTorch version to screen and cleanup (#1325)
* Create flatten_recursive() helper function

* cleanup

* print torch version
2020-11-09 12:24:11 +01:00
Glenn Jocher 81d320109f
--save-txt store_true fix (#1323) 2020-11-08 19:53:48 +01:00
Glenn Jocher 177f39e98e
onnx>=1.8.0 2020-11-08 19:41:54 +01:00
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
2020-11-08 19:39:05 +01:00
Glenn Jocher d3e7778151 imap bug fix 2020-11-07 02:27:31 +01:00
Glenn Jocher 194f16844e Multi-threaded image caching 2020-11-07 02:18:18 +01:00
Glenn Jocher ea7e78cb11
Grid indices overflow bug fix (#1316) 2020-11-06 19:17:50 +01:00
Glenn Jocher 58f8ba771e Creado con Colaboratory 2020-11-06 17:12:23 +01:00
Glenn Jocher fed5d9cdfb
Grid indices overflow bug fix (#1313) 2020-11-06 13:40:11 +01:00
Glenn Jocher bd5cfff8fb Creado con Colaboratory 2020-11-05 20:01:33 +01:00
Glenn Jocher ebe563e701 Created using Colaboratory 2020-11-05 19:47:35 +01:00
Glenn Jocher 9ac9ec3449 Created using Colaboratory 2020-11-05 18:34:15 +01:00
Glenn Jocher 74b34872fd
Add Weights & Biases tutorial (#1290) 2020-11-05 02:30:13 +01:00
Glenn Jocher 15a10609fe
autoshape() update for PIL greyscale inputs (#1279)
* autoshape update for PIL greyscale inputs

* autoshape update for PIL greyscale inputs
2020-11-04 14:20:11 +01:00
Junghoon Kim 187f7c2ed1
Change optimizer parameters group method (#1239)
* Change optimizer parameters group method

* Add torch nn

* Change isinstance method(torch.Tensor to nn.Parameter)

* parameter freeze fix, PEP8 reformat

* freeze bug fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-02 00:08:36 +01:00
Ayush Chaurasia 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
Glenn Jocher 2062765e06
auto-download v3.1 models (#1259) 2020-11-01 20:11:41 +01:00
Ayush Chaurasia 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
Glenn Jocher c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates (#1233)
* torch 1.7.0 compatibility updates

* add inference verification
2020-10-28 15:03:50 +01:00
Glenn Jocher 453acdec67 Update tensorboard logging 2020-10-26 01:08:33 +01:00
Glenn Jocher 7f1640695b
Update hubconf.py (#1210) 2020-10-26 01:06:54 +01:00
Glenn Jocher 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
oleg 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
Glenn Jocher 0a52ae16dc
README Update (#1207)
* README Update

* Update README.md

* README Update

* Update README.md
2020-10-25 14:05:28 +01:00
Glenn Jocher 8e9b3d0205
Precision-Recall Curve feature update #1206 2020-10-25 12:59:16 +01:00
Glenn Jocher ed85038c38
Precision-Recall Curve feature update (#1206)
* Precision-Recall Curve feature update

* sentinel value update
2020-10-25 12:55:52 +01:00
Glenn Jocher 7fe4a6bc06 FROM nvcr.io/nvidia/pytorch:20.10-py3 2020-10-25 10:49:33 +01:00
Glenn Jocher 481d46cffb Improved corruption handling during scan and cache (#999) 2020-10-24 15:09:19 +02:00
Glenn Jocher d61930e017 Improved corruption handling during scan and cache (#999) 2020-10-24 14:50:50 +02:00
Glenn Jocher 0fda95aaf4 Update cache out-of-date warning 2020-10-23 15:13:42 +02:00
Osama Akhtar 83deec133d
Add confidence score to label txt files (#994)
* add contrast to conf/class against bbox rectangle color

* added feature to save conf in txt labels

* Update general.py

* Update detect.py

* Update detect.py

* Update detect.py

save_conf variable removed as only used once, so we can use opt.save_conf instead.

* Update detect.py

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-20 14:20:27 +02:00
Yi_Zhang f532bc68a9
GhostConv() bug fix (#1176) 2020-10-20 13:42:11 +02:00
Shiwei Song 9291daa6f7
fix padding for rectangular inference (#1165)
Co-authored-by: swsong <swsong@stratosphere.mobi>
2020-10-19 12:22:22 +02:00
Glenn Jocher 3b57cb5641
Simplified inference (#1153) 2020-10-15 20:10:08 +02:00
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>
2020-10-15 15:05:58 +02:00
Glenn Jocher 4d3680c81d
Minor import and spelling updates (#1133) 2020-10-13 17:24:27 +02:00
Glenn Jocher 4346b13a40
Dataset download bash script updates (#1132) 2020-10-13 15:58:03 +02:00
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>
2020-10-13 14:10:21 +02:00
Glenn Jocher 0ada058f63
Generalized regression criterion renaming (#1120) 2020-10-11 17:25:17 +02:00
Glenn Jocher 10c85bf4eb
FROM nvcr.io/nvidia/pytorch:20.09-py3 2020-10-11 16:25:14 +02:00
Glenn Jocher d45e349a1e
Rearrange export input after checks (#1118)
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
2020-10-11 16:23:36 +02:00