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

1167 Commits

Author SHA1 Message Date
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
Glenn Jocher
89655a84f2 .fuse() gradient introduction bug fix 2020-09-20 11:57:19 -07:00
Glenn Jocher
5a9c5c1d3b add mosaic and warmup to hyperparameters (#931) 2020-09-13 14:08:06 -07:00
Glenn Jocher
f1c63e2784 add mosaic and warmup to hyperparameters (#931) 2020-09-13 14:03:54 -07:00
Glenn Jocher
806e75f2b1 self.label_files bug fix (#958) 2020-09-13 11:38:23 -07:00
Glenn Jocher
c54e39464c
check_file() assert unique 2020-09-11 14:33:06 -07:00
Glenn Jocher
720645d96a LoadImagesAndLabels() /images/ to /labels/ fix (#928) 2020-09-10 12:29:05 -07:00
Glenn Jocher
d49c52eee3 _RepeatSampler outside of InfiniteDataLoader 2020-09-10 12:27:35 -07:00
Glenn Jocher
9c0efdde9f glob.glob(recursive=True) search (#917) 2020-09-06 22:52:28 -07:00
Glenn Jocher
bc1fd13a7a gsutil cp hyp evolution bug fix (#876) 2020-09-03 20:29:08 -07:00
Glenn Jocher
44cdcc7e0b hyp['anchors'] evolution update 2020-09-03 12:54:22 -07:00
Glenn Jocher
ffe9eb4238 remove if platform.system() == 'Darwin' checks 2020-09-03 11:41:33 -07:00
tkianai
c5969f79f5
add support for DWConv fuse operation (#898) 
* the value should be a list to unpacking

* filter the None labels

* update readme to remove apex dependency

* add support for DWConv fuse operation
 2020-09-01 22:18:42 -07:00
Glenn Jocher
d3f9bf2bb7 Update datasets.py 2020-09-01 17:02:47 -07:00
NanoCode012
1e15aad6f9
Add InfiniteDataLoader class (#876) 
* Add InfiniteDataLoader

Only initializes at first epoch. Saves time.

* Moved class to a better location
 2020-08-31 11:01:25 -07:00
Glenn Jocher
987c226849 seaborn plotting correlogram addition 2020-08-29 15:13:58 -07:00
rafale77
9776e70988
torch.ops.torchvision.nms (#860) 
Don't load the entire torchvision library just for nms when the function is already in the torch library.
 2020-08-27 11:13:41 -07:00
Glenn Jocher
31d6c389f2 Update general.py plot_evolution() 2020-08-25 23:08:14 -07:00
Willie Maddox
455f7b8f76
Update general.py (#823) 
Fixes #822

`init_seeds` from `torch_utils` import is being overwritten by function `init_seeds` in `general.py`
 2020-08-25 13:01:17 -07:00
Glenn Jocher
5a7d79fbe6
Update general.py bbox_iou() with eps improvements (#736) 
eps reduced to 1e-9 for additional precaution.
 2020-08-25 12:56:39 -07:00