Hugging Face's logo

Spaces:
henry000
/
YOLO
Running

App Files Community
YOLO / yolo /tools /data_loader.py

Commit History

Fix coco style bounding box annotations (#116)
8e564ea
unverified

Ramon2 commited on

🔀 [Merge] branch 'MODEL' into TEST
fa2f0be

henry000 commited on

🔀 [Merge] branch 'TRAIN' into TEST
2ab865c

henry000 commited on

🩹 [Fix] config for auto download dataset
564b976

henry000 commited on

🚸 [Add] try-except in loading cache files
1690354

henry000 commited on

[113] - Handle batches with no gt, Add images with no gt to the dataset (#120)
5c767f2
unverified

DmitriiSavin dsavin commited on

🔨 [Add] image ratio min-max constrain
5e0d5e2

henry000 commited on

🐛 [Fix] bugs in dynamic shape in training
3092710

henry000 commited on

✨ [Add] dynamic image size loader
aba5422

henry000 commited on

✏️ [Revert] typo in cache files
b0c4448

henry000 commited on

✏️ [Fix] Process <- Proccess typo
abc3992

henry000 commited on

🐛 [Fix] display on DDP mode
91a3f78

henry000 commited on

♻️ [Refactor] dataset, apply tensorlize
ed7ee89

henry000 commited on

🔀 [Merge] remote-tracking branch 'origin/DDP_BUGS' into Lightning
f789a66

henry000 commited on

🌐 [Add] logging of optimizre(learning rate)
5fcc6be

henry000 commited on

🐛 [Fix] PIL read image, manually close image
ca2b494

henry000 commited on

🗃️ [Update] dataloader, remove YOLOLoader allin1
95520e9

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

🚛 [Move] Tensorlize out of YoloDataset
e09ff86

henry000 commited on

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

henry000 commited on

🐛 [Add] Tensorlize @ dataloader
70a1547

henry000 commited on

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

henry000 commited on

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

henry000 commited on

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

henry000 commited on

✅ [Pass] test with lower system requirements
2ee7407

henry000 commited on

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

henry000 commited on

💬 [Add] TODO reduce memory usage when cal loss
66f3f62

henry000 commited on

🐛 [Add] dynamic bbox, prevent more cuda memory bug
1528a4f

henry000 commited on

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

henry000 commited on

🔨 [Update] dataloader, return data augment info
8b1b21f

henry000 commited on

🔨 [Update] dataloader with reversable padding size
15f0a98

henry000 commited on

🚀 [New] DDP mode for training model
1fe2937

henry000 commited on

🎨 [Update] the dataset check, only check using set
5958998

henry000 commited on

✅ [Pass] git action with install cv2 lib
07fa01d

henry000 commited on

♻️ [Refactor] move to_tensor to transform
7daf6f0

henry000 commited on

🔧 [Update] all structure of yaml, more readability
97e9dcb

henry000 commited on

✨ [Add] A stream dataloader for webcam
8ca39dc

henry000 commited on

🔒️ [Update] Dataset, can output data without label
1ff7fa6

henry000 commited on

🧱 [Update] config files struct, make it clearly
b5fa3f1

henry000 commited on

🚚 [Rename] All utils, tools, higher readability
dcceddd

henry000 commited on