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
Glenn Jocher
d87cf7ecb9
Update requirements.txt coremltools==4.0
2020-10-11 16:21:39 +02:00
Glenn Jocher
c2403d785c
fuse() bug fix
2020-10-11 00:42:54 +02:00
Glenn Jocher
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
Glenn Jocher
9eae82e3a3
Update stale.yml exempt-issue-labels bug fix ( #1098 )
2020-10-08 16:41:04 +02:00
NanoCode012
77940c3f42
Remove ignore for git files ( #1099 )
2020-10-08 15:07:27 +02:00
Frederik
d11504aee1
Remove image open on macos ( #1086 )
...
* added option to not open on save
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-07 16:16:10 +02:00
Glenn Jocher
5b8cf247af
venv* ignore update
2020-10-06 15:17:12 +02:00
Glenn Jocher
124f0e8212
torchvision nms bug fix
2020-10-06 15:09:24 +02:00
Glenn Jocher
66676eb039
init_torch_seeds >> init_seeds bug fix
2020-10-06 15:00:47 +02:00
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
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
2020-10-05 15:46:32 +02:00
Glenn Jocher
35fe98543c
Update requirements.txt coremltools==4.0b4
2020-10-05 01:47:04 +02:00