Tensorboard integration - Githubissues

tinyvision / DAMO-YOLO

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.

Apache License 2.0

3.75k stars 470 forks source link

Tensorboard integration #134

Open ksv87 opened 6 months ago

ksv87 commented 6 months ago

Search before asking

[X] I have searched the DAMO-YOLO issues and found no similar feature requests.

Description

Integration with tensorboard

from torch.utils.tensorboard import SummaryWriter
...
writer.add_scalar(...)
...

https://pytorch.org/docs/stable/tensorboard.html

Use case

This is more informative than a text log

Additional

Map metrics must be included in the tensorboard

Are you willing to submit a PR?

[ ] Yes I'd like to help by submitting a PR!