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年前
Glenn Jocher
d3e7778151
imap bug fix
4年前
Glenn Jocher
194f16844e
Multi-threaded image caching
4年前
Glenn Jocher
ea7e78cb11
Grid indices overflow bug fix ( #1316 )
4年前
Glenn Jocher
fed5d9cdfb
Grid indices overflow bug fix ( #1313 )
4年前
Glenn Jocher
2062765e06
auto-download v3.1 models ( #1259 )
4年前
Glenn Jocher
c8c5ef36c9
PyTorch 1.7.0 Compatibility Updates ( #1233 )
* torch 1.7.0 compatibility updates
* add inference verification
4年前
Glenn Jocher
453acdec67
Update tensorboard logging
4年前
Glenn Jocher
8e9b3d0205
Precision-Recall Curve feature update #1206
4年前
Glenn Jocher
ed85038c38
Precision-Recall Curve feature update ( #1206 )
* Precision-Recall Curve feature update
* sentinel value update
4年前
Glenn Jocher
481d46cffb
Improved corruption handling during scan and cache ( #999 )
4年前
Glenn Jocher
d61930e017
Improved corruption handling during scan and cache ( #999 )
4年前
Shiwei Song
9291daa6f7
fix padding for rectangular inference ( #1165 )
Co-authored-by: swsong <swsong@stratosphere.mobi>
4年前
Glenn Jocher
3b57cb5641
Simplified inference ( #1153 )
4年前
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年前
Glenn Jocher
4d3680c81d
Minor import and spelling updates ( #1133 )
4年前
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年前
Glenn Jocher
0ada058f63
Generalized regression criterion renaming ( #1120 )
4年前
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年前
Glenn Jocher
124f0e8212
torchvision nms bug fix
4年前
Glenn Jocher
883924d9dc
classifier, export, torch seed updates
4年前
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年前
haoyuefan
7220cee1d1
mosaic4 bottom left image fix ( #1021 )
fix a bug in load_mosaic
4年前
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年前
Glenn Jocher
89655a84f2
.fuse() gradient introduction bug fix
4年前
Glenn Jocher
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
4年前
Glenn Jocher
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
4年前
Glenn Jocher
806e75f2b1
self.label_files bug fix ( #958 )
4年前
Glenn Jocher
c54e39464c
check_file() assert unique
4年前
Glenn Jocher
720645d96a
LoadImagesAndLabels() /images/ to /labels/ fix ( #928 )
4年前
Glenn Jocher
d49c52eee3
_RepeatSampler outside of InfiniteDataLoader
4年前
Glenn Jocher
9c0efdde9f
glob.glob(recursive=True) search ( #917 )
4年前
Glenn Jocher
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
4年前
Glenn Jocher
44cdcc7e0b
hyp['anchors'] evolution update
4年前
Glenn Jocher
ffe9eb4238
remove if platform.system() == 'Darwin' checks
4年前
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年前
Glenn Jocher
d3f9bf2bb7
Update datasets.py
4年前
NanoCode012
1e15aad6f9
Add InfiniteDataLoader class ( #876 )
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
4年前
Glenn Jocher
987c226849
seaborn plotting correlogram addition
4年前
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.
4年前
Glenn Jocher
31d6c389f2
Update general.py plot_evolution()
4年前
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`
4年前
Glenn Jocher
5a7d79fbe6
Update general.py bbox_iou() with eps improvements ( #736 )
eps reduced to 1e-9 for additional precaution.
4年前
Glenn Jocher
82ed33a052
generalize webcam detection with opt.source.isnumeric() (yolov3 #1462 )
4年前
Glenn Jocher
4d7f222f73
Update export.py with v3.0 Hardswish() support
4年前
Glenn Jocher
8666bc507e
Create evolve.sh for infinite hyp evolution
4年前
Glenn Jocher
71209a6099
exportable Hardswish() implementation
4年前
Glenn Jocher
f7e075fe93
gsutil cp yaml when evolving to GCP bucket
4年前
Glenn Jocher
9dbaa1bbbe
Update plot_results()
4年前
Glenn Jocher
ea5e97fb9b
FP16 NMS ( #814 )
4年前