marvin/CNN_Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed import bug

Browse Source
This commit is contained in:
marvin
2026-05-31 22:21:48 +02:00
parent edf4f76c6f
commit 89a341750c
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/src/main.py
-1
View File
@@ -1,7 +1,6 @@
import sys import sys
sys.path.insert(1, './src/yolo') sys.path.insert(1, './src/yolo')
from yolo_model_production import convert_prediction, Yolo_model
from train_yolo_faces import train_yolo from train_yolo_faces import train_yolo
sys.path.insert(2, './src/bird_cnn') sys.path.insert(2, './src/bird_cnn')
api/src/yolo/train_yolo_faces.py
+1 -1
View File
@@ -12,7 +12,7 @@ import torch
from torchvision.utils import draw_bounding_boxes from torchvision.utils import draw_bounding_boxes
import torch.nn.functional as F 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): def view_data(dataset):
dataloader = DataLoader(dataset=dataset, batch_size=1, shuffle=False) dataloader = DataLoader(dataset=dataset, batch_size=1, shuffle=False)