Detection support (#60)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
This commit is contained in:
Ayush Chaurasia 2022-12-03 21:01:23 +05:30 committed by GitHub
parent 5a52e7663a
commit 7ec7cf3aef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 545 additions and 11 deletions

View file

@ -250,7 +250,7 @@ class SegmentationTrainer(BaseTrainer):
cls,
bboxes,
masks,
paths,
paths=paths,
fname=self.save_dir / f"train_batch{ni}.jpg")
def plot_metrics(self):