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
2,780 Commits 10 Branches 10 Tags
Commit Graph

1136 Commits

Author SHA1 Message Date
Glenn Jocher
0822cda781 Update caching (#1496) 2020-11-24 16:22:02 +01:00
Glenn Jocher
89c7a5b8dc
Update caching (#1496) 2020-11-24 16:13:04 +01:00
igornishka
44f42b1589
changed prints to logging in utils/datasets (#1315) 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-11-24 16:03:19 +01:00
Glenn Jocher
7aeef2dca5
Prevent PR plotting (#1489) 2020-11-24 00:46:01 +01:00
Glenn Jocher
354109c54c Autosplit (#1488) 2020-11-23 18:35:25 +01:00
Glenn Jocher
4798e66fdf
Autosplit (#1488) 2020-11-23 17:18:21 +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
394131c2aa
Use torchvision.ops.nms (#1460) 2020-11-20 11:23:36 +01:00
yujun
05a955a3f6
FLOPS computation device bug fix (#1447) 
* Update torch_utils.py

fix issue#113 , inputs device should be same with model parameters' device

* Update torch_utils.py

* Update torch_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-11-19 12:56:20 +01:00
Glenn Jocher
64bce3c822 Update labels.png with rectangles fix (#1432) 2020-11-18 12:27:30 +01:00
Glenn Jocher
1542ccadb2
Update labels.png with rectangles (#1432) 2020-11-18 00:17:33 +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
Jacobsolawetz
8d2d6d2349
Update img2label_paths() (#1414) 2020-11-17 13:26:06 +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
92c9b72832
Matplotlib AGG backend (#1413) 2020-11-16 16:34:07 +01:00
Glenn Jocher
548a98a391
Recursive directories dataset capability (#1408) 
* Recursive directories dataset capability

* x.split('.')[-1]

* f += glob.glob(f"{p}{os.sep}**.*", recursive=True)

* f += glob.glob(str(p / '**.*'), recursive=True)

* remove . from image and vid formats

* .txt to txt

* str(p / '**' / '*.*')
 2020-11-16 16:24:57 +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
9f27902368 Improved FLOPS computation (#1398) 2020-11-14 14:48:55 +01:00
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation (#1398) 
* Improved FLOPS computation

* update comment
 2020-11-14 14:39:46 +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
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
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
Glenn Jocher
d3e7778151 imap bug fix 2020-11-07 02:27:31 +01:00
Glenn Jocher
194f16844e Multi-threaded image caching 2020-11-07 02:18:18 +01:00
Glenn Jocher
ea7e78cb11
Grid indices overflow bug fix (#1316) 2020-11-06 19:17:50 +01:00
Glenn Jocher
fed5d9cdfb
Grid indices overflow bug fix (#1313) 2020-11-06 13:40:11 +01:00
Glenn Jocher
2062765e06
auto-download v3.1 models (#1259) 2020-11-01 20:11:41 +01:00
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates (#1233) 
* torch 1.7.0 compatibility updates

* add inference verification
 2020-10-28 15:03:50 +01:00
Glenn Jocher
453acdec67 Update tensorboard logging 2020-10-26 01:08:33 +01:00
Glenn Jocher
8e9b3d0205
Precision-Recall Curve feature update #1206 2020-10-25 12:59:16 +01:00
Glenn Jocher
ed85038c38
Precision-Recall Curve feature update (#1206) 
* Precision-Recall Curve feature update

* sentinel value update
 2020-10-25 12:55:52 +01:00
Glenn Jocher
481d46cffb Improved corruption handling during scan and cache (#999) 2020-10-24 15:09:19 +02:00
Glenn Jocher
d61930e017 Improved corruption handling during scan and cache (#999) 2020-10-24 14:50:50 +02:00
Shiwei Song
9291daa6f7
fix padding for rectangular inference (#1165) 
Co-authored-by: swsong <swsong@stratosphere.mobi>
 2020-10-19 12:22:22 +02:00
Glenn Jocher
3b57cb5641
Simplified inference (#1153) 2020-10-15 20:10:08 +02:00
Jirka Borovec
c67e72200e
fix compatibility for hyper config (#1146) 
* fix/hyper

* Hyp giou check to train.py

* restore general.py

* train.py overwrite fix

* restore general.py and pep8 update

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-10-15 15:05:58 +02:00
Glenn Jocher
4d3680c81d
Minor import and spelling updates (#1133) 2020-10-13 17:24:27 +02:00
Jirka Borovec
00917a6225
update expt name comment and folder parsing for training (#978) 
* comment

* fix parsing

* fix evolve

* folder

* tqdm

* Update train.py

* Update train.py

* reinstate anchors into meta dict

anchor evolution is working correctly now

* reinstate logger

prefer the single line readout for concise logging, which helps simplify notebook and tutorials etc.

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2020-10-13 14:10:21 +02: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
124f0e8212 torchvision nms bug fix 2020-10-06 15:09:24 +02:00
Glenn Jocher
883924d9dc classifier, export, torch seed updates 2020-10-06 14:54:02 +02:00
Jiacong Fang
c5d2331897
Fix increment_dir to use run_xxx for logdir (#1058) 
* Fix increment_dir to use run_xxx for logdir 

Rerunning train.py with logdir in the form of `run_xxx` causes index slicing in
```
n = max([int(x[len(dir):x.find('_') if '_' in Path(x).name else None]) for x in d]) + 1  # increment
```
to underflow.

* Replace find with rfind
 2020-10-05 15:46:32 +02:00
haoyuefan
7220cee1d1
mosaic4 bottom left image fix (#1021) 
fix a bug in load_mosaic
 2020-09-23 09:51:33 -07:00
Guilhen
702c4fa53e
adding the configuration to deploy yolov5 in in app engine (#964) 
* add the configuration to deploy yolov5 in a docker container in app engine on google cloud

* remove the duplicated requirements, place the google app engine files in utils

* remove the unused requirements

Co-authored-by: guilhembau <guilhem@6dbytes.com>
 2020-09-22 11:48:44 -07:00