fixed import bug

2026-05-31 22:21:48 +02:00
parent edf4f76c6f
commit 89a341750c
2 changed files with 1 additions and 2 deletions
@@ -12,7 +12,7 @@ import torch
 from torchvision.utils import draw_bounding_boxes
 import torch.nn.functional as F

-from yolo_model_production import convert_prediction
+from production.yolo_model_production import convert_prediction

 def view_data(dataset):
    dataloader = DataLoader(dataset=dataset, batch_size=1, shuffle=False)