Jacobsolawetz
8d2d6d2349
Update img2label_paths() ( #1414 )
4 years ago
Glenn Jocher
f5429260ca
PyTorch Hub and autoShape update ( #1415 )
* PyTorch Hub and autoShape update
* comment x for imgs
* reduce comment
4 years ago
Glenn Jocher
92c9b72832
Matplotlib AGG backend ( #1413 )
4 years ago
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 / '**' / '*.*')
4 years ago
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
4 years ago
Glenn Jocher
9f27902368
Improved FLOPS computation ( #1398 )
4 years ago
Glenn Jocher
d8f5fcfe87
Improved FLOPS computation ( #1398 )
* Improved FLOPS computation
* update comment
4 years ago
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 years ago
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
4 years ago
Glenn Jocher
67bf9a974e
Replace 'ground truth' with 'labels' ( #1337 )
* Replace 'ground truth' with 'labels'
* Using torch... update
4 years ago
Glenn Jocher
19e2482458
PyTorch version to screen and cleanup ( #1325 )
* Create flatten_recursive() helper function
* cleanup
* print torch version
4 years ago
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
4 years ago
Glenn Jocher
d3e7778151
imap bug fix
4 years ago
Glenn Jocher
194f16844e
Multi-threaded image caching
4 years ago
Glenn Jocher
ea7e78cb11
Grid indices overflow bug fix ( #1316 )
4 years ago
Glenn Jocher
fed5d9cdfb
Grid indices overflow bug fix ( #1313 )
4 years ago
Glenn Jocher
2062765e06
auto-download v3.1 models ( #1259 )
4 years ago
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4 years ago
Glenn Jocher
453acdec67
Update tensorboard logging
4 years ago
Glenn Jocher
8e9b3d0205
Precision-Recall Curve feature update #1206
4 years ago
Glenn Jocher
ed85038c38
Precision-Recall Curve feature update ( #1206 )
* Precision-Recall Curve feature update
* sentinel value update
4 years ago
Glenn Jocher
481d46cffb
Improved corruption handling during scan and cache ( #999 )
4 years ago
Glenn Jocher
d61930e017
Improved corruption handling during scan and cache ( #999 )
4 years ago
Shiwei Song
9291daa6f7
fix padding for rectangular inference ( #1165 )
Co-authored-by: swsong <swsong@stratosphere.mobi>
4 years ago
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4 years ago
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>
4 years ago
Glenn Jocher
4d3680c81d
Minor import and spelling updates ( #1133 )
4 years ago
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>
4 years ago
Glenn Jocher
0ada058f63
Generalized regression criterion renaming ( #1120 )
4 years ago
Glenn Jocher
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
4 years ago
Glenn Jocher
124f0e8212
torchvision nms bug fix
4 years ago
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
4 years ago
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
4 years ago
haoyuefan
7220cee1d1
mosaic4 bottom left image fix ( #1021 )
fix a bug in load_mosaic
4 years ago
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>
4 years ago
Glenn Jocher
89655a84f2
.fuse() gradient introduction bug fix
4 years ago
Glenn Jocher
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
4 years ago
Glenn Jocher
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
4 years ago
Glenn Jocher
806e75f2b1
self.label_files bug fix ( #958 )
4 years ago
Glenn Jocher
c54e39464c
check_file() assert unique
4 years ago
Glenn Jocher
720645d96a
LoadImagesAndLabels() /images/ to /labels/ fix ( #928 )
4 years ago
Glenn Jocher
d49c52eee3
_RepeatSampler outside of InfiniteDataLoader
4 years ago
Glenn Jocher
9c0efdde9f
glob.glob(recursive=True) search ( #917 )
4 years ago
Glenn Jocher
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
4 years ago
Glenn Jocher
44cdcc7e0b
hyp['anchors'] evolution update
4 years ago
Glenn Jocher
ffe9eb4238
remove if platform.system() == 'Darwin' checks
4 years ago
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
4 years ago
Glenn Jocher
d3f9bf2bb7
Update datasets.py
4 years ago
NanoCode012
1e15aad6f9
Add InfiniteDataLoader class ( #876 )
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
4 years ago
Glenn Jocher
987c226849
seaborn plotting correlogram addition
4 years ago