Glenn Jocher
9f27902368
Improved FLOPS computation ( #1398 )
4 years ago
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
4 years ago
Glenn Jocher
0c26c4e831
Increase pycocotools robustness ( #1396 )
4 years ago
Glenn Jocher
9b0f6e33ee
Increase pycocotools robustness ( #1396 )
4 years ago
Glenn Jocher
cf581305db
Creado con Colaboratory
4 years ago
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 years ago
Glenn Jocher
379396e896
Yaml constructor posixpath --resume bug fix ( #1390 )
* resume fix for yaml constructor posixpath error
* fix update
* remove weights/ dir backup
4 years ago
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 years ago
weida wang
04081f8102
Trailing --save-txt whitespace bug fix ( #1374 )
* update Write results
* Update detect.py
* Update test.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
Glenn Jocher
ce3872f86c
W&B evolution bug fix ( #1373 )
4 years ago
Glenn Jocher
f6b3c96966
Update tutorial.ipynb
4 years ago
Glenn Jocher
9cd46427e4
Creado con Colaboratory
4 years ago
Glenn Jocher
104f5418ea
Creado con Colaboratory
4 years ago
Glenn Jocher
7841d7ba88
Update README.md
4 years ago
Glenn Jocher
6ff0d916fb
Update download_weights.sh ( #1338 )
4 years ago
Glenn Jocher
27138c2ebb
Update download_weights.sh ( #1338 )
4 years ago
Glenn Jocher
67bf9a974e
Replace 'ground truth' with 'labels' ( #1337 )
* Replace 'ground truth' with 'labels'
* Using torch... update
4 years ago
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>
4 years ago
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
4 years ago
Glenn Jocher
81d320109f
--save-txt store_true fix ( #1323 )
4 years ago
Glenn Jocher
177f39e98e
onnx>=1.8.0
4 years ago
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 years ago
Glenn Jocher
d3e7778151
imap bug fix
4 years ago
Glenn Jocher
194f16844e
Multi-threaded image caching
4 years ago
Glenn Jocher
ea7e78cb11
Grid indices overflow bug fix ( #1316 )
4 years ago
Glenn Jocher
58f8ba771e
Creado con Colaboratory
4 years ago
Glenn Jocher
fed5d9cdfb
Grid indices overflow bug fix ( #1313 )
4 years ago
Glenn Jocher
bd5cfff8fb
Creado con Colaboratory
4 years ago
Glenn Jocher
ebe563e701
Created using Colaboratory
4 years ago
Glenn Jocher
9ac9ec3449
Created using Colaboratory
4 years ago
Glenn Jocher
74b34872fd
Add Weights & Biases tutorial ( #1290 )
4 years ago
Glenn Jocher
15a10609fe
autoshape() update for PIL greyscale inputs ( #1279 )
* autoshape update for PIL greyscale inputs
* autoshape update for PIL greyscale inputs
4 years ago
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>
4 years ago
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>
4 years ago
Glenn Jocher
2062765e06
auto-download v3.1 models ( #1259 )
4 years ago
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>
4 years ago
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4 years ago
Glenn Jocher
453acdec67
Update tensorboard logging
4 years ago
Glenn Jocher
7f1640695b
Update hubconf.py ( #1210 )
4 years ago
Glenn Jocher
d3dad42256
Move generated *.json annotations into runs/test ( #1209 )
* Move generated *.json annotations into runs/test
* list and string robust
4 years ago
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>
4 years ago
Glenn Jocher
0a52ae16dc
README Update ( #1207 )
* README Update
* Update README.md
* README Update
* Update README.md
4 years ago
Glenn Jocher
8e9b3d0205
Precision-Recall Curve feature update #1206
4 years ago
Glenn Jocher
ed85038c38
Precision-Recall Curve feature update ( #1206 )
* Precision-Recall Curve feature update
* sentinel value update
4 years ago
Glenn Jocher
7fe4a6bc06
FROM nvcr.io/nvidia/pytorch:20.10-py3
4 years ago
Glenn Jocher
481d46cffb
Improved corruption handling during scan and cache ( #999 )
4 years ago
Glenn Jocher
d61930e017
Improved corruption handling during scan and cache ( #999 )
4 years ago
Glenn Jocher
0fda95aaf4
Update cache out-of-date warning
4 years ago
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>
4 years ago
Yi_Zhang
f532bc68a9
GhostConv() bug fix ( #1176 )
4 years ago