Commit History

✨ [New] Add a classification head!
d44dbc0

henry000 commited on

πŸ”€ [Merge] branch 'DDP_BUGS' of github.com:WongKinYiu/yolov9mit into DDP_BUGS
b9867bb

henry000 commited on

πŸ› [Fix] Loss scale, scale back with batch_size
22ebde1

henry000 commited on

πŸš› [Move] Tensorlize out of YoloDataset
e09ff86

henry000 commited on

πŸ› [Fix] DDP bugs, sync mAP on diff rank
5be4e0e

henry000 commited on

πŸ› [Fix] bbox_nms function (#76)
44527a6
unverified

f-fl0 commited on

πŸ› [Update] getitem with removing -1 cls padding
e64ae7a

henry000 commited on

✏️ [Fix] typo in README doc
8228669

henry000 commited on

πŸ”€ [Merge] branch 'main' into SETUP
86cd350

henry000 commited on

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

henry000 commited on

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

henry000 commited on

✍️ [Update] README.md
a7ca20f
unverified

HenryTsui commited on

πŸ› [Add] Tensorlize @ dataloader
70a1547

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] bugs, reduce test py version, lower cache
0bd2faa

henry000 commited on

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

henry000 commited on

πŸ”€ [Merge] branch SETUP into TEST
b8b460c

henry000 commited on

πŸ“ [Update] Project badge, add two state icon
f739434

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] Pull request template
648dce4

henry000 commited on

πŸ“„ [Update] Pull Requst template, mentioned by #51
efc2687

henry000 commited on

πŸ”€ [Merge] branch 'SETUP'
fb4930b

henry000 commited on

πŸ’š [Fix] test bugs, pass streamloader orgimg shape
1e4ec0a

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

πŸ› [Fix] #56 bugs, create_converter -> Vec2Box
054a1c7

henry000 commited on

πŸ“ [Update] pretrained weight of v7
d2fb889

henry000 commited on

πŸ› [Fix] requirement of docs, add tabs ext
6bac4af

henry000 commited on

πŸ“ [Add] function document (by autodoc)
0141cf8

henry000 commited on

πŸ“ [Add] TODO title in custom section
b5b053e

henry000 commited on

πŸ“ [Update] document of get start, turn table->tabs
0f23de9

henry000 commited on

πŸ“ [Update] tutorial, split cfg, data, model, task
4a690bb

henry000 commited on