f532bc68a9
GhostConv() bug fix ( #1176 )
2020-10-20 13:42:11 +02:00
9291daa6f7
fix padding for rectangular inference ( #1165 )
...
Co-authored-by: swsong <swsong@stratosphere.mobi>
2020-10-19 12:22:22 +02:00
3b57cb5641
Simplified inference ( #1153 )
2020-10-15 20:10:08 +02:00
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
4d3680c81d
Minor import and spelling updates ( #1133 )
2020-10-13 17:24:27 +02:00
4346b13a40
Dataset download bash script updates ( #1132 )
2020-10-13 15:58:03 +02:00
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
0ada058f63
Generalized regression criterion renaming ( #1120 )
2020-10-11 17:25:17 +02:00
10c85bf4eb
FROM nvcr.io/nvidia/pytorch:20.09-py3
2020-10-11 16:25:14 +02:00
d45e349a1e
Rearrange export input after checks ( #1118 )
...
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
2020-10-11 16:23:36 +02:00
d87cf7ecb9
Update requirements.txt coremltools==4.0
2020-10-11 16:21:39 +02:00
c2403d785c
fuse() bug fix
2020-10-11 00:42:54 +02:00
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
9eae82e3a3
Update stale.yml exempt-issue-labels bug fix ( #1098 )
2020-10-08 16:41:04 +02:00
77940c3f42
Remove ignore for git files ( #1099 )
2020-10-08 15:07:27 +02:00
d11504aee1
Remove image open on macos ( #1086 )
...
* added option to not open on save
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-10-07 16:16:10 +02:00
5b8cf247af
venv* ignore update
2020-10-06 15:17:12 +02:00
124f0e8212
torchvision nms bug fix
2020-10-06 15:09:24 +02:00
66676eb039
init_torch_seeds >> init_seeds bug fix
2020-10-06 15:00:47 +02:00
883924d9dc
classifier, export, torch seed updates
2020-10-06 14:54:02 +02:00
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
35fe98543c
Update requirements.txt coremltools==4.0b4
2020-10-05 01:47:04 +02:00
2d2371d2d1
Update get_voc.sh data download URLs ( #1077 )
2020-10-04 19:35:05 +02:00
a12698f4c9
Update ci-testing.yml reflecting sys.path additions ( #869 )
...
* Update ci-testing.yml
* Update yolo.py
* Update ci-testing.yml
* Update export.py
* Update yolo.py
2020-10-04 18:50:32 +02:00
d55c326dda
export.py folder sys.path addition ( #1052 )
2020-10-04 17:42:05 +02:00
7220cee1d1
mosaic4 bottom left image fix ( #1021 )
...
fix a bug in load_mosaic
2020-09-23 09:51:33 -07:00
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
b8f656742d
feat: add rtmp support ( #1009 )
...
* feat: add rtmp support
* Update detect.py
pass tuple to source.startswith()
Co-authored-by: ryan.fu <ryan.fu@aylaasia.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-09-22 11:37:49 -07:00
4fce0096fa
model.add_nms() method
2020-09-20 16:31:40 -07:00
2f77cf33f6
.fuse() additional error checking
2020-09-20 12:04:20 -07:00
89655a84f2
.fuse() gradient introduction bug fix
2020-09-20 11:57:19 -07:00
c4cb78570c
add NMS to pretrained pytorch hub models
2020-09-18 18:17:11 -07:00
5a9c5c1d3b
add mosaic and warmup to hyperparameters ( #931 )
2020-09-13 14:08:06 -07:00
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
2020-09-13 14:03:54 -07:00
806e75f2b1
self.label_files bug fix ( #958 )
2020-09-13 11:38:23 -07:00
6062bb7dee
print self.yaml['nc'] override update
2020-09-11 17:44:11 -07:00
a62a45b2dd
prevent testloader caching on --notest
2020-09-11 16:59:13 -07:00
35dc935504
remove `nn.Conv2d` in default settings ( #949 )
2020-09-11 16:35:48 -07:00
c54e39464c
check_file() assert unique
2020-09-11 14:33:06 -07:00
720645d96a
LoadImagesAndLabels() /images/ to /labels/ fix ( #928 )
2020-09-10 12:29:05 -07:00
d49c52eee3
_RepeatSampler outside of InfiniteDataLoader
2020-09-10 12:27:35 -07:00
bb8872ea5f
Dockerfile sudo docker "kill all" bug fix
2020-09-08 00:56:35 -07:00
cab36f72a8
add .git to .dockerignore
2020-09-06 22:53:12 -07:00
9c0efdde9f
glob.glob(recursive=True) search ( #917 )
2020-09-06 22:52:28 -07:00
c8e51812a5
hyp evolution force-autoanchor fix
2020-09-04 13:13:10 -07:00
c687d5c129
reorganize train initialization steps
2020-09-04 12:25:53 -07:00
bc1fd13a7a
gsutil cp hyp evolution bug fix ( #876 )
2020-09-03 20:29:08 -07:00
44cdcc7e0b
hyp['anchors'] evolution update
2020-09-03 12:54:22 -07:00
ffe9eb4238
remove if platform.system() == 'Darwin' checks
2020-09-03 11:41:33 -07:00
23a7f29e2d
Update hyp.finetune.yaml on 306 VOC generations result
2020-09-03 01:20:30 -07:00
Yi_Zhang
Shiwei Song
Glenn Jocher
Jirka Borovec
NanoCode012
Frederik
Jiacong Fang
Mingyu Wang
PetrDvoracek
haoyuefan
Guilhen
stone99
tomguluson92