mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
924 Commits 10 Branches 10 Tags
Commit Graph

106 Commits

Author SHA1 Message Date
Glenn Jocher
e27ca0d845
Update minimum stride to 32 (#2266) 2021-02-21 21:46:42 -08:00
Glenn Jocher
c09964c27c
Update inference default to multi_label=False (#2252) 
* Update inference default to multi_label=False

* bug fix

* Update plots.py

* Update plots.py
 2021-02-19 15:39:09 -08:00
Glenn Jocher
be9edffded
Update test.py --task speed and study (#2099) 
* Add --speed benchmark

* test range 256 - 1536

* update

* update

* update

* update
 2021-01-31 13:58:49 -08:00
Glenn Jocher
f639e14e4d
Metric-Confidence plots feature addition (#2057) 
* Metric-Confidence plots feature addition

* cleanup

* Metric-Confidence plots feature addition

* cleanup

* Update run-once lines

* cleanup

* save all 4 curves to wandb
 2021-01-27 17:10:53 -08:00
Glenn Jocher
2a835c79a9
Update run-once lines (#2058) 2021-01-27 16:01:24 -08:00
Glenn Jocher
59c21c7bcb
W&B log epoch (#1946) 
* W&B log epoch

* capitalize

* W&B log epoch

* capitalize

* Update train.py

New try using https://docs.wandb.ai/library/log#incremental-logging

* Update train.py

* Update test.py

* Update train.py

* Update plots.py

* Update train.py

* Update train.py

* label plot step -1

* update

* update

* update

* update

* update

* update

* Update train.py

* Update train.py
 2021-01-26 21:16:01 -08:00
ramonhollands
d68afedb32
Confusion matrix native image-space fix (#2046) 
Make sure the labels and predictions are equally scaled on confusion_matrix.process_batch
 2021-01-26 13:39:19 -08:00
Glenn Jocher
046c37e465
verbose on final_epoch (#1997) 2021-01-20 18:27:38 -08:00
Abhiram V
91c30e4eff
Update test.py with yaml.SafeLoader (#1969) 2021-01-18 10:46:46 -08:00
Glenn Jocher
ca9babb8e6
Add ComputeLoss() class (#1950) 2021-01-15 13:50:24 -08:00
Glenn Jocher
dd03b20ba5
colorstr() updates (#1909) 
* W&B ImportError message fix

* colorstr() updates

* colorstr() updates

* colorstr() default to 'blue', 'bold'

* train: magenta

* train: blue
 2021-01-12 10:33:15 -08:00
Glenn Jocher
3e25f1e9e5
auto-verbose if nc <=20 (#1869) 2021-01-07 15:47:28 -08:00
Glenn Jocher
e77c77f580
Add check_requirements() (#1853) 
* Add check_requirements()

* add import

* parameterize filename

* add to detect, test
 2021-01-06 16:35:40 -08:00
Glenn Jocher
86f4247515
Hybrid auto-labelling support (#1646) 2020-12-08 18:15:39 -08:00
Glenn Jocher
791dadb51c
Pycocotools best.pt after COCO train (#1616) 
* Pycocotools best.pt after COCO train

* cleanup
 2020-12-06 14:58:33 +01:00
Glenn Jocher
b6ed1104a6
Daemon thread plotting (#1561) 
* Daemon thread plotting

* remove process_batch

* plot after print
 2020-11-30 16:44:14 +01:00
Glenn Jocher
0f2057ed33
Targets scaling bug fix (#1529) 2020-11-26 18:33:28 +01:00
Glenn Jocher
e9a0ae6f19
Cache bug fix (#1513) 
* Caching bug fix #1508

* np.zeros((0,5)) x2
 2020-11-25 20:33:14 +01:00
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
 2020-11-23 15:27:14 +01:00
Glenn Jocher
95fa65339f
Cat apriori to autolabels (#1484) 2020-11-23 13:38:47 +01:00
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
 2020-11-21 12:38:35 +01:00
Glenn Jocher
a70e554fc3
Remove redundant downloads mirror (#1461) 2020-11-20 13:10:56 +01:00
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()
 2020-11-18 22:50:21 +01:00
Glenn Jocher
412b113ac6 W&B logging bug fix (#1399) 2020-11-17 19:17:57 +01:00
Glenn Jocher
4250f84dfb
Update PR curve (#1428) 
* Update PR curve

* legend outside

* list(Path().glob())
 2020-11-17 17:29:55 +01:00
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update (#1415) 
* PyTorch Hub and autoShape update

* comment x for imgs

* reduce comment
 2020-11-16 23:09:55 +01:00
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
 2020-11-16 13:35:34 +01:00
Glenn Jocher
0c26c4e831 Increase pycocotools robustness (#1396) 2020-11-14 13:53:58 +01:00
Glenn Jocher
9b0f6e33ee
Increase pycocotools robustness (#1396) 2020-11-14 13:09:04 +01:00
Glenn Jocher
fe341fa44d
Utils reorganization (#1392) 
* Utils reorganization

* Add new utils files

* cleanup

* simplify

* reduce datasets.py

* remove evolve.sh

* loadWebcam cleanup
 2020-11-14 11:50:32 +01:00
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
 2020-11-12 23:37:46 +01:00
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>
 2020-11-12 17:37:16 +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
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
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
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
453acdec67 Update tensorboard logging 2020-10-26 01:08:33 +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
0ada058f63
Generalized regression criterion renaming (#1120) 2020-10-11 17:25:17 +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
c171e45cda Update test.py with set() (#806) 2020-08-21 15:03:03 -07:00
Glenn Jocher
da35727c68
--save-txt bug fix (#763) 2020-08-17 17:14:07 -07:00
Glenn Jocher
f5da528d28 reformat code 2020-08-14 11:53:44 -07:00
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
 2020-08-13 14:25:05 -07:00
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>
 2020-08-12 14:18:19 -07:00
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
 2020-08-09 20:52:57 -07:00
Jirka Borovec
d5b6416c87
Explicit Imports (#498) 
* expand imports

* optimize

* miss

* fix
 2020-08-02 15:47:36 -07:00