changed imports

This commit is contained in:
2026-05-21 05:49:48 +02:00
parent a519e77db1
commit 640c852e80
5 changed files with 8 additions and 18 deletions
+1 -10
View File
@@ -1,21 +1,12 @@
import json
import os
import cv2
from matplotlib import pyplot as plt
import numpy as np
from torch import tensor
import torch
from torch.utils.data import DataLoader
from torchvision import transforms
from torchvision.transforms import ToTensor, functional
import torchvision.transforms.functional as TF
from tqdm import tqdm
from torchvision.transforms import ToTensor
import torch.nn as nn
import torch.nn.functional as F
from .r_cnn import ObjectDetectionCNN, train, eval
from .cocoDetectionDataset import CocoDetectionDataset
import random
SAVE_PATH = "./saved_models"
PADDING = 20