 Glenn Jocher
|
61ea23c3fe
|
Implement `@torch.no_grad()` decorator (#3312)
* `@torch.no_grad()` decorator
* Update detect.py
|
пре 3 година |
 kepler62f
|
f2de1ad2aa
|
Comment fix (#3058)
Co-authored-by: kepler62f <>
|
пре 3 година |
|
Glenn Jocher
|
a833ee2a46
|
Update check_requirements() exclude list (#2974)
|
пре 3 година |
|
Glenn Jocher
|
3665c0f59b
|
test.py native --single-cls (#2928)
|
пре 3 година |
 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
|
пре 3 година |
|
Glenn Jocher
|
f7bc685c2c
|
Implement yaml.safe_load() (#2876)
* Implement yaml.safe_load()
* yaml.safe_dump()
|
пре 3 година |
 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>
|
пре 3 година |
 Benjamin Fineran
|
fd1679975b
|
add option to disable half precision in test.py (#2507)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
пре 3 година |
 Ayush Chaurasia
|
8f6e447729
|
Fix Indentation in test.py (#2614)
* Fix Indentation in test.py
* CI fix
* Comply with PEP8: 80 characters per line
|
пре 3 година |
|
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>
|
пре 3 година |
|
Glenn Jocher
|
f4197214aa
|
Update test.py --task train val study (#2453)
* Update test.py --task train val study
* update argparser --task
|
пре 3 година |
|
Glenn Jocher
|
f01f3223d5
|
Integer printout (#2450)
* Integer printout
* test.py 'Labels'
* Update train.py
|
пре 3 година |
|
Glenn Jocher
|
cd30d838eb
|
Update test.py (#2319)
|
пре 3 година |
|
Glenn Jocher
|
71dd2768f2
|
Improved model+EMA checkpointing 2 (#2295)
|
пре 3 година |
|
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
|
пре 3 година |
|
Glenn Jocher
|
e27ca0d845
|
Update minimum stride to 32 (#2266)
|
пре 3 година |
|
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
|
пре 3 година |
|
Glenn Jocher
|
be9edffded
|
Update test.py --task speed and study (#2099)
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
|
пре 3 година |
|
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
|
пре 3 година |
|
Glenn Jocher
|
2a835c79a9
|
Update run-once lines (#2058)
|
пре 3 година |
|
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
|
пре 3 година |
 ramonhollands
|
d68afedb32
|
Confusion matrix native image-space fix (#2046)
Make sure the labels and predictions are equally scaled on confusion_matrix.process_batch
|
пре 3 година |
|
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 година |
