YOLO / yolo /tools /solver.py

Commit History

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

Martin Cerman mcerman-enlite 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

✨ [Add] mutli-GPU on validation mode
e0c8580

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

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

henry000 commited on

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

henry000 commited on

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

henry000 commited on

✏️ [Fix] typo in loading coco gt files
ce70a60

henry000 commited on

πŸ”¨ [Update] AMP, unscale gradient and update model
192b787

henry000 commited on

⚑️ [Update] Optimizer, with more fine LR
5b9da41

henry000 commited on

βœ… [Fix] some path bug and enable ignore run pycoco
b038f54

henry000 commited on

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

henry000 commited on

✨ [Add] saving function when higher mAP
73b88fc

henry000 commited on

βœ… [Add] tests, increase test coverage
88e45b9

henry000 commited on

πŸ› [Fix] Some display bug, enable print, log
46c9d1c

henry000 commited on

πŸ’¬ [Update] logging, add avg loss and ap table
c3b133e

henry000 commited on

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

henry000 commited on

✨ [New] inference with coco format
999037c

henry000 commited on

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

henry000 commited on

πŸ”€ [Merge] branch 'INFERENCE' into DEPLOY
2222e19

henry000 commited on

πŸ”€ [Merge] branch 'INFERENCE' into TEST
f5a3a55

henry000 commited on

πŸš€ [Update] the postproccess -> PostProccess class
329fd0a

henry000 commited on

✨ [New] validation code! run my pycocotools
3441a79

henry000 commited on

πŸ”¨ [Update] dataloader, return data augment info
8b1b21f

henry000 commited on

🎨 [Update] the progress bar, with xx -> xx.start()
2275731

henry000 commited on

🎨 [Update] drawer input, more robust/consistency
64acfd1

henry000 commited on

πŸ’¬ [Update] logging tools for quite mode
afa32b4

henry000 commited on

πŸš€ [New] DDP mode for training model
1fe2937

henry000 commited on

⚑️ [Update] using anchor if is given, or autoanchor
41f1f41

henry000 commited on

πŸ› [Fix] some bugs, fit the create_model, device
78e3679

henry000 commited on

πŸ”€ [Merge] branch 'SETUP' into MODELv2
682daad

henry000 commited on

:speech_balloon: [Add] FPS when using stream data
a0976c9

henry000 commited on

:truck: [Rename] Tracker to Logger, handle all log
d58a9b6

henry000 commited on

:bug: [Fix] a bug with drawing picture after merge
635f41a

henry000 commited on

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

henry000 commited on

[🎨] Add support for displaying webcam videos with predicted bounding boxes (#27)
ecf6aba
unverified

LaiEthanLai HenryTsui commited on

βœ… [Pass] the test and rename the function of v2 arch
d1477fc

henry000 commited on

πŸ› [Update] some bug or vaiable name in Vec2Box
f95a3d7

henry000 commited on

✨ [New] Framework anc2box -> anc2vec + vec2box
2dd2ae5

henry000 commited on

πŸ”₯ [Remove] scale function in drawer
3b24b35

henry000 commited on

✨ [Add] New model, anchor2box move into model
86ef0ef

henry000 commited on

πŸ’¬ [Update] Bbox drawing tools, change label text
7692528

henry000 commited on

✨ [New] validation code!
b2baf14

henry000 commited on

πŸ”§ [Update] all structure of yaml, more readability
97e9dcb

henry000 commited on

πŸ”§ [Update] default confidence score & error handle
bbd2c43

henry000 commited on

♻️ [Refactor] the code when inference
f3e770a

henry000 commited on