issue in YOLOv8Detector
roykev opened this issue · comments
hi,
I'm running the detector on a custom model (i.e., i provide the the weights and classes)
got an error in line 107
warnings.warn(
f"class {_class} not found in model classes list.")
I can resolve it when I replace line 99
class_names = get_names()
with
class_names = list(self.model.names.values())
Am I right?
thanks