Glenn Jocher
|
2268f9c44f
|
update *.yaml output layer comments
|
2020-07-30 11:31:47 -07:00 |
Liu Changyu
|
adaf1d8ac1
|
Update models/hub *.yaml files for v2.0 (#546)
* fix #543 issue
* Update yolov3-spp.yaml
* Update yolov5-fpn.yaml
* Update yolov5-panet.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2020-07-29 14:02:07 -07:00 |
|
Glenn Jocher
|
43a616a955
|
PyTorch 1.6.0 compatability updates
|
2020-07-28 15:38:56 -07:00 |
|
Glenn Jocher
|
1d17b9af0f
|
update yolo.py TTA flexibility and extensibility (#506)
* update yolo.py TTA flexibility and extensibility
* Update scale_img()
|
2020-07-24 11:42:23 -07:00 |
|
Glenn Jocher
|
9da56b62dd
|
v2.0 Release (#491)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
2020-07-23 15:34:23 -07:00 |
|
Glenn Jocher
|
a97c3f94ec
|
update common.py Classify()
|
2020-07-16 23:59:51 -07:00 |
|
Glenn Jocher
|
5387d4747d
|
update common.py add Classify()
|
2020-07-16 17:18:41 -07:00 |
|
Glenn Jocher
|
2b1892430f
|
yolo.py bug fix
|
2020-07-13 15:21:19 -07:00 |
|
Glenn Jocher
|
02445d176d
|
improved model.yaml source tracking
|
2020-07-13 14:35:47 -07:00 |
|
Glenn Jocher
|
9006b85d21
|
update export.py torchscript file extension
|
2020-07-10 11:56:01 -07:00 |
|
Glenn Jocher
|
16f6834486
|
update train.py and experimental.py
|
2020-07-08 14:23:34 -07:00 |
Laurentiu Diaconu
|
a9918fbf2b
|
updated coreml conversion to do pixel scaling between 0.0 and 1.0
|
2020-07-08 09:43:33 +03:00 |
|
Glenn Jocher
|
e8cf24b6c8
|
Initial model ensemble capability #318
|
2020-07-07 15:40:50 -07:00 |
|
Glenn Jocher
|
04bdbe4104
|
fuse update
|
2020-07-05 23:16:50 -07:00 |
|
Glenn Jocher
|
5ba1de0cdc
|
update experimental.py with Ensemble() module
|
2020-07-05 15:02:56 -07:00 |
|
Glenn Jocher
|
38f5c1ad1d
|
pruning and sparsity initial commit
|
2020-07-05 13:41:21 -07:00 |
|
Glenn Jocher
|
997ba7b346
|
import yaml in yolo.py
|
2020-07-05 12:50:04 -07:00 |
|
Glenn Jocher
|
762b06de4b
|
Merge remote-tracking branch 'origin/master'
|
2020-07-04 23:07:32 -07:00 |
|
Glenn Jocher
|
4e2d24602d
|
update yolo.py
|
2020-07-04 23:07:26 -07:00 |
|
Glenn Jocher
|
bb3c346916
|
model.yaml nc inherited from dataset.yaml
|
2020-07-04 17:51:54 -07:00 |
|
Glenn Jocher
|
659ad748c8
|
update get_voc.sh
|
2020-07-04 17:13:43 -07:00 |
|
Glenn Jocher
|
7bd7b2c090
|
update export.py
|
2020-07-03 20:05:50 -07:00 |
|
Glenn Jocher
|
655895a838
|
module updates
|
2020-07-03 13:46:12 -07:00 |
|
Glenn Jocher
|
a9d20eba3e
|
export.py update
|
2020-07-03 11:50:59 -07:00 |
|
Glenn Jocher
|
569fbc63c8
|
export.py update
|
2020-07-03 11:29:53 -07:00 |
|
Glenn Jocher
|
a03adb5ab8
|
Merge remote-tracking branch 'origin/master'
|
2020-07-01 19:16:06 -07:00 |
|
Glenn Jocher
|
1fca7a7f24
|
autopad() update in common.py
|
2020-07-01 19:15:59 -07:00 |
|
Glenn Jocher
|
a62a1c2c67
|
export.py update
|
2020-07-01 16:14:49 -07:00 |
|
Glenn Jocher
|
f1d67f4110
|
update export.py
|
2020-07-01 15:46:15 -07:00 |
|
Glenn Jocher
|
b5659d1195
|
module updates
|
2020-07-01 11:44:49 -07:00 |
|
Glenn Jocher
|
615d6d0cfa
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
2020-06-29 14:00:13 -07:00 |
|
Glenn Jocher
|
e8bf385795
|
Merge pull request #223 from ArmyOfRobots/master
Adding torchscript export
|
2020-06-29 13:39:53 -07:00 |
|
Glenn Jocher
|
0bc80e1e57
|
onnx_export.py bug fix #224
|
2020-06-28 17:07:03 -07:00 |
Jake Poznanski
|
685091a3fc
|
Adding torchscript export
|
2020-06-28 13:17:05 -07:00 |
|
Glenn Jocher
|
9a9c4f1259
|
update yolo.py default --cfg
|
2020-06-24 18:00:43 -07:00 |
|
Glenn Jocher
|
e071b8dd3a
|
update models/common.py for Conv() flexible padding
|
2020-06-24 13:12:23 -07:00 |
|
Glenn Jocher
|
0825cb7fd8
|
create models/hub
|
2020-06-24 11:58:23 -07:00 |
|
Glenn Jocher
|
b1b3634145
|
update onnx_export.py #134
|
2020-06-24 11:00:03 -07:00 |
|
Glenn Jocher
|
8699c319ca
|
update yolo.py
|
2020-06-24 10:42:01 -07:00 |
|
Glenn Jocher
|
53294f28b4
|
update yolo.py
|
2020-06-23 10:53:42 -07:00 |
|
Glenn Jocher
|
364fcfd7db
|
PANet update
|
2020-06-22 14:27:17 -07:00 |
|
Glenn Jocher
|
fc171e2678
|
check_anchor_order() update
|
2020-06-21 14:08:59 -07:00 |
|
Glenn Jocher
|
e249792c2f
|
update onnx_export.py usage
|
2020-06-19 10:15:08 -07:00 |
|
Glenn Jocher
|
80b82e8bde
|
update yolo.py
|
2020-06-18 19:33:02 -07:00 |
|
Glenn Jocher
|
e1e3399286
|
ONNX export bug fix #93
|
2020-06-16 09:59:42 -07:00 |
ashing
|
62eda5e9c2
|
add "__init__.py" file in <models> folder
|
2020-06-15 14:37:07 +08:00 |
|
Glenn Jocher
|
4ba47b1b1b
|
model.yaml updates
|
2020-06-12 15:41:46 -07:00 |
|
Glenn Jocher
|
1e2cb6b480
|
ONNX export explicit cpu map_location
|
2020-06-12 13:54:00 -07:00 |
|
Glenn Jocher
|
8f17a96ff0
|
cleanup
|
2020-06-11 21:55:22 -07:00 |
|
Glenn Jocher
|
8f1755290c
|
ONNX export update
|
2020-06-11 21:54:01 -07:00 |