YOLO / yolo

Commit History

🩹 [Fix] BoxMatcher, change eps and filter function
fd5413f

henry000 commited on

✨ [Update] back the warm up batch step
604c897

henry000 commited on

πŸ”¨ [Update] gradient scaler
32405d5

henry000 commited on

πŸ› [Fix] bug in DDP sync_dist
fc29438

henry000 commited on

πŸ”’οΈ [Fix] torch version diff in meshgrid
593a50f

henry000 commited on

πŸ”Š [Add] supporting of tensorboard
de99a93

henry000 commited on

πŸ—ƒοΈ [Update] dataloader, remove YOLOLoader allin1
95520e9

henry000 commited on

πŸš‘οΈ [Fix] progress bar in DDP or multiGPU env
1d2b161

henry000 commited on

πŸ”Š [Update] log item, format value on wandb
240dcb0

henry000 commited on

πŸ› [Update] Momentum of the optimizer
42e909b

henry000 commited on

✨ [New] use lightning framework to training!
8b3b3ef

henry000 commited on

πŸ› [Fix] display emoji bugs, change it to shortcode
802cb12

henry000 commited on

πŸ› [Fix] Dataloader log error, remove format str
3d0f0de

henry000 commited on

πŸ’¬ [Update] logging -> loguru, consistency logging
0174b5b

henry000 commited on

πŸ”¨ [FIX] Fixes memory leak (#83)
dea5a8a
unverified

Martin Cerman mcerman-enlite commited on

⬆️ [Update] dependency and the load cache function
699f2d2

henry000 commited on

⬆️ [Update] mAP from 0~1 to percentage
0efc848

henry000 commited on

πŸ› [Fix] mul-GPU @ valid bugs, return mAP @ rk 0
a44a5c1

henry000 commited on

πŸ› [Fix] #38 local mAP bugs, one pd match < 1 gt
aee6ac8

henry000 commited on

✨ [Add] mutli-GPU on validation mode
e0c8580

henry000 commited on

πŸ› [Fix] LR schedule bugs, batchidx should increase
10420ed

henry000 commited on

🎨 [Fix] logging bugs, only log on rank0 gpu
ed7d166

henry000 commited on

πŸ‘½οΈ [Update] cudaamp, most torch not support amp yet
360e5d7

henry000 commited on

πŸ› [Fix] loss, avg loss if using ddp
3b24cc8

henry000 commited on

🚨 [Update] cuda.amp to amp, fix torch warning
9fc0d0e

henry000 commited on

πŸ”€ [Merge] branch 'DATASET' into TRAIN
46b5efe

henry000 commited on

♻️ [Refactor] the dataloader import and default aug
8fc6449

henry000 commited on

🦺 [Add] a RemoveOutlier handle outside bbox
489f14b

henry000 commited on

✨ [Add] a RandomCrop transform, align paper
c627401

henry000 commited on

πŸš‘οΈ [Fix] bbox bugs, choice 1 st bbox when draw
dc88787

henry000 commited on

🦺 [Fix] draw image bbox bugs, make it robust
47093ac

henry000 commited on

✨ [Add] Visualize image after validation, train
9c191b9

henry000 commited on

✨ [Add] TensorBoard support, finish #52
7f8fc3e

henry000 commited on

πŸ“ [Update] wandb logger name
7e9db93

henry000 commited on

♻️ [Refactor] the tf board name, change lower case
f4a5047

henry000 commited on

♻️ [Refactor] create_model function
0ff0fd0

henry000 commited on

πŸ”€ [Merge] branch 'DEPLOY' into TEST
5c8e313

henry000 commited on

🚚 [Rename] class, fix #49 issue, it's not repN
8a2b98c

henry000 commited on

🦺 [Update] load weight, can load trained weight
6972568

henry000 commited on

πŸ› [Fix] #56 bugs, the using converter in ipynb
f1bfd74

henry000 commited on

πŸ”€ [Merge] branch 'DATASET' into TEST
868c821

henry000 commited on

πŸ”€ [Merge] branch 'MODEL' into TEST
9c96910

henry000 commited on

🚚 [Update] the class_num&class_list to dataset
3eb85fd

henry000 commited on

πŸ”’οΈ πŸ”’οΈ [Update] dataset, can also load bbox GT
d2d621f

henry000 commited on

πŸ”’οΈ [Update] safe load model, handle mismatch weight
4775b4c

henry000 commited on

πŸ”¨ [Remove] device in loss func, not sure correct
60c4943

henry000 commited on

πŸ’„ [Update] logging function, log only rank0 device
1132d27

henry000 commited on

πŸ”€ [Merge] branch 'TRAIN' into TEST
ba6baa6

henry000 commited on

πŸ”€ [Merge] branch 'MODEL' into TEST
877244a

henry000 commited on

πŸ”€ [Merge] branch 'DEPLOY' into TEST
814608f

henry000 commited on