Commits · henry000/YOLO

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

0efc848

henry000 commited on Aug 14, 2024

🐛 [Fix] #38 local mAP bugs, one pd match < 1 gt

aee6ac8

henry000 commited on Aug 10, 2024

✨ [Add] mutli-GPU on validation mode

e0c8580

henry000 commited on Aug 10, 2024

🐛 [Fix] LR schedule bugs, batchidx should increase

10420ed

henry000 commited on Aug 10, 2024

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

ed7d166

henry000 commited on Aug 9, 2024

🔀 [Merge] branch 'DATASET' into TRAIN

46b5efe

henry000 commited on Aug 9, 2024

♻️ [Refactor] the dataloader import and default aug

8fc6449

henry000 commited on Aug 9, 2024

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

47093ac

henry000 commited on Aug 5, 2024

✨ [Add] Visualize image after validation, train

9c191b9

henry000 commited on Aug 5, 2024

✨ [Add] TensorBoard support, finish #52

7f8fc3e

henry000 commited on Aug 5, 2024

📝 [Update] wandb logger name

7e9db93

henry000 commited on Aug 5, 2024

🔀 [Merge] branch 'DATASET' into TEST

868c821

henry000 commited on Jul 24, 2024

🚚 [Update] the class_num&class_list to dataset

3eb85fd

henry000 commited on Jul 24, 2024

🔒️ 🔒️ [Update] dataset, can also load bbox GT

d2d621f

henry000 commited on Jul 23, 2024

🔨 [Remove] device in loss func, not sure correct

60c4943

henry000 commited on Jul 23, 2024

💄 [Update] logging function, log only rank0 device

1132d27

henry000 commited on Jul 23, 2024

🔀 [Merge] branch 'TRAIN' into TEST

ba6baa6

henry000 commited on Jul 22, 2024

🎨 [Add] type hint for converter, Anc2Box/Vec2Box

c289952

henry000 commited on Jul 22, 2024

⚡️ [Update] Optimizer, with more fine LR

5b9da41

henry000 commited on Jul 22, 2024

🔧 [Add] const random seed

598b827

henry000 commited on Jul 22, 2024

🐛 [Fix] bugs in onnx's output and video source

100c13d

henry000 commited on Jul 18, 2024

💚 [Fix] MPS CI bugs, enable using MPS

4056352

henry000 commited on Jul 10, 2024

✅ [Pass] test with fixup choosing device

ecc08c8

henry000 commited on Jul 10, 2024

🚑️ [Fix] get_device typo, not return cpu

3b31306

henry000 commited on Jul 10, 2024

♻️ [Refactor] the code of v7 converter, align v9

323161f

henry000 commited on Jul 8, 2024

🚸 [Add] converter function, auto choose conveter

0d49177

henry000 commited on Jul 7, 2024

♻️ [Refactor] Vec2Box, Align with YOLOv7 structure

fd35390

henry000 commited on Jul 6, 2024

🔨 [Add] Anc2Box conveter, for YOLOv7's output

d5a73bd

henry000 commited on Jul 6, 2024

🔒️ [Add] type for logging model, capable for YOLO

68d5954

henry000 commited on Jul 1, 2024

🔀 Merge branch 'TRAIN' into TEST

35562e8

henry000 commited on Jun 22, 2024

🌱 [Add] a random seed for training

67c056a

henry000 commited on Jun 22, 2024

✅ [Add] test, increase test coverage for dev mode

6a39ae1

henry000 commited on Jun 22, 2024

✅ [Pass] test with lower system requirements

2ee7407

henry000 commited on Jun 21, 2024

✅ [Fix] some path bug and enable ignore run pycoco

b038f54

henry000 commited on Jun 20, 2024

♻️ [Refactor] os.path -> pathlib Path, more safely

fa09d11

henry000 commited on Jun 20, 2024

✨ [Add] saving function when higher mAP

73b88fc

henry000 commited on Jun 20, 2024

✅ [Add] tests, increase test coverage

88e45b9

henry000 commited on Jun 19, 2024

🐛 [Fix] onnx deplying bug

7a45a10

henry000 commited on Jun 19, 2024

🐛 [Fix] Some display bug, enable print, log

46c9d1c

henry000 commited on Jun 19, 2024

🔀 [Merge] branch 'TRAIN' into TEST

1ded329

henry000 commited on Jun 19, 2024

🐛 [Fix] #32 bugs, turn 8400->auto anchor size

7f8ebc5

henry000 commited on Jun 19, 2024

💬 [Update] logging, add avg loss and ap table

c3b133e

henry000 commited on Jun 19, 2024

✨ [New] local mAP calculation, AP.5:.95 - mAP.5

21a413f

henry000 commited on Jun 18, 2024