xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Clean and bump dvc callback: settings, stacked images (#4343)

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ivan Shcheklein
2023-08-13 12:50:35 -07:00
committed by GitHub
parent b5d1af42d8
commit d704507217
4 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/engine/trainer.py
View File

@ -543,7 +543,8 @@ class BaseTrainer:
def on_plot(self, name, data=None):
"""Registers plots (e.g. to be consumed in callbacks)"""
self.plots[name] = {'data': data, 'timestamp': time.time()}
path = Path(name)
self.plots[path] = {'data': data, 'timestamp': time.time()}
def final_eval(self):
"""Performs final evaluation and validation for object detection YOLO model."""

3
ultralytics/engine/validator.py
View File

@ -303,7 +303,8 @@ class BaseValidator:
def on_plot(self, name, data=None):
"""Registers plots (e.g. to be consumed in callbacks)"""
self.plots[name] = {'data': data, 'timestamp': time.time()}
path = Path(name)
self.plots[path] = {'data': data, 'timestamp': time.time()}
# TODO: may need to put these following functions into callback
def plot_val_samples(self, batch, ni):