changed yolo loss

2026-05-22 05:57:40 +02:00
parent e412738d3e
commit b6a4dd5fa2
5 changed files with 40 additions and 32 deletions
@@ -4,7 +4,7 @@ import torch
 import torchvision.transforms.functional as TF
 from tqdm import tqdm
 import torch.nn as nn
-from src.util import iou, visualizeImage
+from util import iou, visualizeImage
 import random
 import torch.nn.functional as F