Glenn Jocher
|
13a09fa68a
|
Update datasets.py
|
5 years ago |
|
Glenn Jocher
|
47bf1730fa
|
Update datasets.py
|
5 years ago |
|
Glenn Jocher
|
f9bab6b127
|
Update datasets.py
|
5 years ago |
lorenzomammana
|
54a9e4f876
|
Refactor code to reduce duplication
|
5 years ago |
|
lorenzomammana
|
94342acbe1
|
Handle multiple datasets
|
5 years ago |
|
Glenn Jocher
|
bf6f41567a
|
hyperparameter printout update
|
5 years ago |
|
Glenn Jocher
|
dc5e18390a
|
Merge branch 'master' into advanced_logging
|
5 years ago |
|
Glenn Jocher
|
cbe39a1dbb
|
Update utils.py
|
5 years ago |
|
Glenn Jocher
|
c654d18f00
|
Update utils.py
|
5 years ago |
|
Glenn Jocher
|
31b07b3550
|
Merge pull request #275 from wanghaoyang0106/master
[bug fix] potential problem if img fed to model is in rectangular shape
|
5 years ago |
|
Glenn Jocher
|
e8cf24b6c8
|
Initial model ensemble capability #318
|
5 years ago |
Alex Stoken
|
95f0a56df7
|
Bug fix to get_latest_run when recent run is named
|
5 years ago |
|
Alex Stoken
|
f517ba81c7
|
Merge branch 'master' into advanced_logging
|
5 years ago |
|
Glenn Jocher
|
121d90b3f2
|
update fuse_conv_and_bn()
|
5 years ago |
|
Glenn Jocher
|
38f5c1ad1d
|
pruning and sparsity initial commit
|
5 years ago |
NanoCode012
|
ccfa9373ae
|
Update to remove "module" attribute check
|
5 years ago |
|
NanoCode012
|
f83521ab9a
|
Fix save error for multi-gpu
|
5 years ago |
|
Glenn Jocher
|
ce0c58f678
|
update compute_loss()
|
5 years ago |
|
Glenn Jocher
|
af41083d52
|
EMA FP16 fix #279
|
5 years ago |
|
Glenn Jocher
|
f767023c56
|
offset and balance update
|
5 years ago |
|
Glenn Jocher
|
df224a0d8f
|
EMA bug fix #279
|
5 years ago |
wanghaoyang0106
|
b57f83d005
|
[bug fix] potential problem if img fed to model is in rectangular shape
|
5 years ago |
|
Glenn Jocher
|
f02481c73a
|
Update torch_utils.py
|
5 years ago |
|
Glenn Jocher
|
fc7c42723d
|
Update utils.py
|
5 years ago |
yxNONG
|
53cdaf6bf5
|
Update utils.py
|
5 years ago |
|
Glenn Jocher
|
9a9333d245
|
add replicate() to datasets.py
|
5 years ago |
|
Glenn Jocher
|
37e13f8846
|
update mosaic border
|
5 years ago |
|
Glenn Jocher
|
93a6765806
|
update mosaic border
|
5 years ago |
|
Alex Stoken
|
e18e6811dc
|
Merge branch 'master' into advanced_logging
|
5 years ago |
|
Glenn Jocher
|
77fb8ee082
|
refactor dataloader
|
5 years ago |
|
Glenn Jocher
|
22fb2b0c25
|
refactor dataloader
|
5 years ago |
|
Glenn Jocher
|
b8557f87e3
|
add stride to datasets.py
|
5 years ago |
|
Glenn Jocher
|
3b062254a6
|
update utils.create_pretrained()
|
5 years ago |
|
Glenn Jocher
|
6c1b87a42e
|
update google_utils import
|
5 years ago |
|
Glenn Jocher
|
364fcfd7db
|
PANet update
|
5 years ago |
|
Glenn Jocher
|
fc171e2678
|
check_anchor_order() update
|
5 years ago |
|
Glenn Jocher
|
1f1917ef56
|
remove fast, add merge
|
5 years ago |
|
Alex Stoken
|
e572bb0803
|
Add plot_results save location to log_dir
|
5 years ago |
|
Glenn Jocher
|
024a42f4fd
|
datasets.py vid_formats updated
|
5 years ago |
|
Glenn Jocher
|
5c2cd711df
|
vid_formats bug fix #146
|
5 years ago |
|
Glenn Jocher
|
10145833c3
|
strip_optimizer() default to save FP16 model
|
5 years ago |
|
Glenn Jocher
|
cce95e744d
|
backbone as FP16, save default to FP32
|
5 years ago |
|
Glenn Jocher
|
9fdb0fbacf
|
AutoAnchor bug fix # 117
|
5 years ago |
|
Alex Stoken
|
945307beba
|
Add save_dir to plot_lr_scheduler and plot_labels
Set save_dir = log_dir in train.py
|
5 years ago |
|
Alex Stoken
|
3263a204ea
|
Fix get_latest_run to search 'search_dir' recursivly
|
5 years ago |
|
Alex Stoken
|
d34291733b
|
Fix get_latest_run() to search inside 'weights' subfolders
|
5 years ago |
|
Glenn Jocher
|
a557b7d924
|
check_img_size() update
|
5 years ago |
|
Glenn Jocher
|
034609414e
|
AutoAnchor fitness to screen
|
5 years ago |
|
Glenn Jocher
|
ef75f7313d
|
AutoAnchor n, imgsz bug fix
|
5 years ago |
|
Glenn Jocher
|
1dfc28527f
|
AutoAnchor BPR to 4 significant figures
|
5 years ago |
|
Glenn Jocher
|
4052603e1f
|
AutoAnchor update - improvement check
|
5 years ago |
|
Glenn Jocher
|
bafbc65ee3
|
AutoAnchor bug fix
|
5 years ago |
|
Alex Stoken
|
25e51bcec7
|
add util function to get most recent last.pt file
added logic in train.py __main__ to handle resuming from a run
|
5 years ago |
|
Alex Stoken
|
490f1e7b9c
|
add save_dir arg to plot_lr_scheduler, default to current dir.
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
|
5 years ago |
|
Glenn Jocher
|
57a0ae3350
|
AutoAnchor implementation
|
5 years ago |
|
Glenn Jocher
|
95c46f7245
|
Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library
Delete redundant thrid party library
|
5 years ago |
|
Glenn Jocher
|
ec81c7b5f2
|
check_anchors() bug fix #90
|
5 years ago |
|
Glenn Jocher
|
8b26e89006
|
AutoAnchor bug fix #72
|
5 years ago |
|
Glenn Jocher
|
8fa3724072
|
kmean_anchors() update
|
5 years ago |
|
Glenn Jocher
|
1c0b6236e3
|
update fast mode
|
5 years ago |
Lornatang
|
8b8b792802
|
There is no need to download extra packages, official bring it with you
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
|
5 years ago |
|
Glenn Jocher
|
db2c3acd3a
|
updated testing settings, rebalanced towards FP16 latency
|
5 years ago |
|
Glenn Jocher
|
bd3e389130
|
update google_utils.py
|
5 years ago |
|
Glenn Jocher
|
c5966abba8
|
glob search bug fix #77
|
5 years ago |
Lornatang
|
54795a4f0a
|
fix problem
|
5 years ago |
|
Glenn Jocher
|
260b1729f0
|
FP16 inference update
|
5 years ago |
|
Glenn Jocher
|
93bbf1b87a
|
best possible recall fix
|
5 years ago |
|
Glenn Jocher
|
31f3310029
|
assert best possible recall > 0.9 before training
|
5 years ago |
|
Glenn Jocher
|
099e6f5ebd
|
--img-size stride-multiple verification
|
5 years ago |
|
Glenn Jocher
|
3a5c5328c5
|
ONNX, BCEBlurWithLogitsLoss, plot_study updates
|
5 years ago |
Trevor Sullivan
|
ac93e9c811
|
Fix Google Drive URI for YOLOv5 weights
|
5 years ago |
|
Glenn Jocher
|
c14368d768
|
new CSP model release
|
5 years ago |
|
Glenn Jocher
|
ff02ae0869
|
module updates
|
5 years ago |
|
Glenn Jocher
|
1f698af6f1
|
create_backbone with SourceChangeWarning update
|
5 years ago |
|
Glenn Jocher
|
8fe299f179
|
model fuse
|
5 years ago |
|
Glenn Jocher
|
b810b21299
|
augmented inference
|
5 years ago |
|
Glenn Jocher
|
c12023f145
|
updates
|
5 years ago |
|
Glenn Jocher
|
66d73e4cf3
|
NMS updates
|
5 years ago |
|
Glenn Jocher
|
948bcdd219
|
--classes bug fix #17
|
5 years ago |
|
Glenn Jocher
|
22d6088205
|
speed-reproducibility fix #17
|
5 years ago |
|
Glenn Jocher
|
0362f103ef
|
updates
|
5 years ago |
|
Glenn Jocher
|
2a0aff685a
|
dataset padding mode
|
5 years ago |
|
Glenn Jocher
|
eb97b2e413
|
NMS fast mode
|
5 years ago |
|
Glenn Jocher
|
ee8988b8a2
|
datasets.py self.batch_shapes fix
|
5 years ago |
|
Glenn Jocher
|
f96e8f3edf
|
updates
|
5 years ago |
|
Glenn Jocher
|
b10609fe6b
|
updates
|
5 years ago |
|
Glenn Jocher
|
72eed215ff
|
updates
|
5 years ago |
|
Glenn Jocher
|
5a79b5f65c
|
updates
|
5 years ago |
|
Glenn Jocher
|
4f44aaf26b
|
updates
|
5 years ago |
|
Glenn Jocher
|
1e84a23f38
|
initial commit
|
5 years ago |
