mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,142 Commits
10 Branches
10 Tags
584 MiB
Commit Graph

121 Commits (c6b5bfca8592c3426ce0b5f65e559c45d42ff378)

Author SHA1 Message Date
Glenn Jocher 61ea23c3fe
Implement `@torch.no_grad()` decorator (#3312) 
* `@torch.no_grad()` decorator

* Update detect.py
 2021-05-24 13:23:09 +02:00
kepler62f f2de1ad2aa
Comment fix (#3058) 
Co-authored-by: kepler62f <>
 2021-05-07 12:28:55 +02:00
Glenn Jocher a833ee2a46
Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
Glenn Jocher 3665c0f59b
test.py native --single-cls (#2928) 2021-04-25 20:05:16 +02:00
fcakyon b40dd99167
Explicit opt function arguments (#2817) 
* more explicit function arguments

* fix typo in detect.py

* revert import order

* revert import order

* remove default value
 2021-04-22 18:17:30 +02:00
Glenn Jocher f7bc685c2c
Implement yaml.safe_load() (#2876) 
* Implement yaml.safe_load()

* yaml.safe_dump()
 2021-04-21 14:34:45 +02:00
Burhan c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub (#2827) 
* Update detect.py

* Update detect.py

* Update greetings.yml

* Update cropping

* cleanup

* Update increment_path()

* Update common.py

* Update detect.py

* Update detect.py

* Update detect.py

* Update common.py

* cleanup

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-04-20 23:51:08 +02:00
Benjamin Fineran fd1679975b
add option to disable half precision in test.py (#2507) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-03-29 17:15:26 +02:00
Ayush Chaurasia 8f6e447729
Fix Indentation in test.py (#2614) 
* Fix Indentation in test.py

* CI fix

* Comply with PEP8: 80 characters per line
 2021-03-26 20:47:11 +01:00
Ayush Chaurasia e8fc97aa38
Improved W&B integration (#2125) 
* Init Commit

* new wandb integration

* Update

* Use data_dict in test

* Updates

* Update: scope of log_img

* Update: scope of log_img

* Update

* Update: Fix logging conditions

* Add tqdm bar, support for .txt dataset format

* Improve Result table Logger

* Init Commit

* new wandb integration

* Update

* Use data_dict in test

* Updates

* Update: scope of log_img

* Update: scope of log_img

* Update

* Update: Fix logging conditions

* Add tqdm bar, support for .txt dataset format

* Improve Result table Logger

* Add dataset creation in training script

* Change scope: self.wandb_run

* Add wandb-artifact:// natively

you can now use --resume with wandb run links

* Add suuport for logging dataset while training

* Cleanup

* Fix: Merge conflict

* Fix: CI tests

* Automatically use wandb config

* Fix: Resume

* Fix: CI

* Enhance: Using val_table

* More resume enhancement

* FIX : CI

* Add alias

* Get useful opt config data

* train.py cleanup

* Cleanup train.py

* more cleanup

* Cleanup| CI fix

* Reformat using PEP8

* FIX:CI

* rebase

* remove uneccesary changes

* remove uneccesary changes

* remove uneccesary changes

* remove unecessary chage from test.py

* FIX: resume from local checkpoint

* FIX:resume

* FIX:resume

* Reformat

* Performance improvement

* Fix local resume

* Fix local resume

* FIX:CI

* Fix: CI

* Imporve image logging

* (:(:Redo CI tests:):)

* Remember epochs when resuming

* Remember epochs when resuming

* Update DDP location

Potential fix for #2405

* PEP8 reformat

* 0.25 confidence threshold

* reset train.py plots syntax to previous

* reset epochs completed syntax to previous

* reset space to previous

* remove brackets

* reset comment to previous

* Update: is_coco check, remove unused code

* Remove redundant print statement

* Remove wandb imports

* remove dsviz logger from test.py

* Remove redundant change from test.py

* remove redundant changes from train.py

* reformat and improvements

* Fix typo

* Add tqdm tqdm progress when scanning files, naming improvements

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-03-23 00:44:50 +01:00
Glenn Jocher f4197214aa
Update test.py --task train val study (#2453) 
* Update test.py --task train val study

* update argparser --task
 2021-03-12 22:08:42 -08:00
Glenn Jocher f01f3223d5
Integer printout (#2450) 
* Integer printout

* test.py 'Labels'

* Update train.py
 2021-03-12 13:35:44 -08:00
Glenn Jocher cd30d838eb
Update test.py (#2319) 2021-02-27 15:28:23 -08:00
Glenn Jocher 71dd2768f2
Improved model+EMA checkpointing 2 (#2295) 2021-02-24 21:03:21 -08:00
Glenn Jocher ec1d8496ba
Improved model+EMA checkpointing (#2292) 
* Enhanced model+EMA checkpointing

* update

* bug fix

* bug fix 2

* always save optimizer

* ema half

* remove model.float()

* model half

* carry ema/model in fp32

* rm model.float()

* both to float always

* cleanup

* cleanup
 2021-02-24 18:26:46 -08:00
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