Glenn Jocher
046c37e465
verbose on final_epoch ( #1997 )
3年前
Abhiram V
91c30e4eff
Update test.py with yaml.SafeLoader ( #1969 )
3年前
Glenn Jocher
ca9babb8e6
Add ComputeLoss() class ( #1950 )
3年前
Glenn Jocher
dd03b20ba5
colorstr() updates ( #1909 )
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
3年前
Glenn Jocher
3e25f1e9e5
auto-verbose if nc <=20 ( #1869 )
3年前
Glenn Jocher
e77c77f580
Add check_requirements() ( #1853 )
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
3年前
Glenn Jocher
86f4247515
Hybrid auto-labelling support ( #1646 )
3年前
Glenn Jocher
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
* Pycocotools best.pt after COCO train
* cleanup
3年前
Glenn Jocher
b6ed1104a6
Daemon thread plotting ( #1561 )
* Daemon thread plotting
* remove process_batch
* plot after print
3年前
Glenn Jocher
0f2057ed33
Targets scaling bug fix ( #1529 )
3年前
Glenn Jocher
e9a0ae6f19
Cache bug fix ( #1513 )
* Caching bug fix #1508
* np.zeros((0,5)) x2
3年前
Glenn Jocher
0a3ff71ae0
Confusion matrix ( #1474 )
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* Cat apriori to autolabels
* cleanup
3年前
Glenn Jocher
95fa65339f
Cat apriori to autolabels ( #1484 )
3年前
Glenn Jocher
201bafc7cf
Sync train and test iou_thresh ( #1465 )
* Sync train and test iou_thresh
* Sync train and test iou_thresh
* weights names .lower()
* Notebook update
3年前
Glenn Jocher
a70e554fc3
Remove redundant downloads mirror ( #1461 )
3年前
Glenn Jocher
225845e781
Update test.py for IoU in native image-space ( #1439 )
* Update test.py for IoU in native image-space
* remove redundant
* gn to device
* remove output scale_coords
* --img-size correction
* update
* native-space labels
* pred to predn
* remove clip_coords()
3年前
Glenn Jocher
412b113ac6
W&B logging bug fix ( #1399 )
3年前
Glenn Jocher
4250f84dfb
Update PR curve ( #1428 )
* Update PR curve
* legend outside
* list(Path().glob())
3年前
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
3年前
Glenn Jocher
9c91aeae10
W&B logging add hyperparameters ( #1399 )
* W&B logging add hyperparameters
* hyp bug fix and image logging updates
* if plots and wandb:
* cleanup
* wandb/ gitignore add
* cleanup 2
* cleanup 3
* move wandb import to top of file
* wandb evolve
* update import
* wandb.run.finish()
* default anchors: 3
3年前
Glenn Jocher
0c26c4e831
Increase pycocotools robustness ( #1396 )
3年前
Glenn Jocher
9b0f6e33ee
Increase pycocotools robustness ( #1396 )
3年前
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
3年前
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
3年前
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>
3年前
Glenn Jocher
67bf9a974e
Replace 'ground truth' with 'labels' ( #1337 )
* Replace 'ground truth' with 'labels'
* Using torch... update
3年前
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>
3年前
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
3年前
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
3年前
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>
3年前
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>
3年前
Glenn Jocher
453acdec67
Update tensorboard logging
3年前
Glenn Jocher
d3dad42256
Move generated *.json annotations into runs/test ( #1209 )
* Move generated *.json annotations into runs/test
* list and string robust
3年前
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>
3年前
Glenn Jocher
0ada058f63
Generalized regression criterion renaming ( #1120 )
4年前
Glenn Jocher
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
4年前
Glenn Jocher
c171e45cda
Update test.py with set() ( #806 )
4年前
Glenn Jocher
da35727c68
--save-txt bug fix ( #763 )
4年前
Glenn Jocher
f5da528d28
reformat code
4年前
Glenn Jocher
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
4年前
NanoCode012
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>
4年前
Glenn Jocher
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
4年前
Jirka Borovec
d5b6416c87
Explicit Imports ( #498 )
* expand imports
* optimize
* miss
* fix
4年前
Glenn Jocher
43a616a955
PyTorch 1.6.0 compatability updates
4年前
Glenn Jocher
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
4年前
Glenn Jocher
b40852d5a5
update test.py
4年前
Jirka
e8ea772384
revert test module to confuse users...
4年前
Jirka
ef6f5b33a8
rename test
4年前
Glenn Jocher
1e831bd820
Update eval.py
4年前
Glenn Jocher
dfb5d27194
Update eval.py
4年前