Glenn Jocher
|
9006b85d21
|
update export.py torchscript file extension
|
4 anni fa |
Glenn Jocher
|
16f6834486
|
update train.py and experimental.py
|
4 anni fa |
Laurentiu Diaconu
|
a9918fbf2b
|
updated coreml conversion to do pixel scaling between 0.0 and 1.0
|
4 anni fa |
Glenn Jocher
|
e8cf24b6c8
|
Initial model ensemble capability #318
|
4 anni fa |
Glenn Jocher
|
04bdbe4104
|
fuse update
|
4 anni fa |
Glenn Jocher
|
5ba1de0cdc
|
update experimental.py with Ensemble() module
|
4 anni fa |
Glenn Jocher
|
38f5c1ad1d
|
pruning and sparsity initial commit
|
4 anni fa |
Glenn Jocher
|
997ba7b346
|
import yaml in yolo.py
|
4 anni fa |
Glenn Jocher
|
4e2d24602d
|
update yolo.py
|
4 anni fa |
Glenn Jocher
|
bb3c346916
|
model.yaml nc inherited from dataset.yaml
|
4 anni fa |
Glenn Jocher
|
659ad748c8
|
update get_voc.sh
|
4 anni fa |
Glenn Jocher
|
7bd7b2c090
|
update export.py
|
4 anni fa |
Glenn Jocher
|
655895a838
|
module updates
|
4 anni fa |
Glenn Jocher
|
a9d20eba3e
|
export.py update
|
4 anni fa |
Glenn Jocher
|
569fbc63c8
|
export.py update
|
4 anni fa |
Glenn Jocher
|
1fca7a7f24
|
autopad() update in common.py
|
4 anni fa |
Glenn Jocher
|
a62a1c2c67
|
export.py update
|
4 anni fa |
Glenn Jocher
|
f1d67f4110
|
update export.py
|
4 anni fa |
Glenn Jocher
|
b5659d1195
|
module updates
|
4 anni fa |
Glenn Jocher
|
615d6d0cfa
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
4 anni fa |
Glenn Jocher
|
0bc80e1e57
|
onnx_export.py bug fix #224
|
4 anni fa |
Jake Poznanski
|
685091a3fc
|
Adding torchscript export
|
4 anni fa |
Glenn Jocher
|
9a9c4f1259
|
update yolo.py default --cfg
|
4 anni fa |
Glenn Jocher
|
e071b8dd3a
|
update models/common.py for Conv() flexible padding
|
4 anni fa |
Glenn Jocher
|
0825cb7fd8
|
create models/hub
|
4 anni fa |
Glenn Jocher
|
b1b3634145
|
update onnx_export.py #134
|
4 anni fa |
Glenn Jocher
|
8699c319ca
|
update yolo.py
|
4 anni fa |
Glenn Jocher
|
53294f28b4
|
update yolo.py
|
4 anni fa |
Glenn Jocher
|
364fcfd7db
|
PANet update
|
4 anni fa |
Glenn Jocher
|
fc171e2678
|
check_anchor_order() update
|
4 anni fa |
Glenn Jocher
|
e249792c2f
|
update onnx_export.py usage
|
4 anni fa |
Glenn Jocher
|
80b82e8bde
|
update yolo.py
|
4 anni fa |
Glenn Jocher
|
e1e3399286
|
ONNX export bug fix #93
|
4 anni fa |
ashing
|
62eda5e9c2
|
add "__init__.py" file in <models> folder
|
4 anni fa |
Glenn Jocher
|
4ba47b1b1b
|
model.yaml updates
|
4 anni fa |
Glenn Jocher
|
1e2cb6b480
|
ONNX export explicit cpu map_location
|
4 anni fa |
Glenn Jocher
|
8f17a96ff0
|
cleanup
|
4 anni fa |
Glenn Jocher
|
8f1755290c
|
ONNX export update
|
4 anni fa |
Glenn Jocher
|
3a5c5328c5
|
ONNX, BCEBlurWithLogitsLoss, plot_study updates
|
4 anni fa |
Glenn Jocher
|
a814720403
|
PyTorch Hub updates
|
4 anni fa |
Glenn Jocher
|
c14368d768
|
new CSP model release
|
4 anni fa |
Glenn Jocher
|
854ccb9a74
|
_print_weights()
|
4 anni fa |
Glenn Jocher
|
5bee686649
|
model definition update
|
4 anni fa |
Glenn Jocher
|
715cb08b10
|
module updates
|
4 anni fa |
Glenn Jocher
|
ff02ae0869
|
module updates
|
4 anni fa |
Glenn Jocher
|
12b0c046d5
|
model fusion and onnx export
|
4 anni fa |
Glenn Jocher
|
b810b21299
|
augmented inference
|
4 anni fa |
Glenn Jocher
|
5c470d235e
|
multi-gpu test bug fix #15
|
4 anni fa |
Glenn Jocher
|
67d5e65c77
|
profiling update
|
4 anni fa |
Glenn Jocher
|
2703ac73e8
|
profiling update
|
4 anni fa |