Skip to content

Instantly share code, notes, and snippets.

@SannaPersson
Created March 21, 2021 10:47
Show Gist options
  • Save SannaPersson/e199c9fb7a44a88b6ec59bf15301e642 to your computer and use it in GitHub Desktop.
Save SannaPersson/e199c9fb7a44a88b6ec59bf15301e642 to your computer and use it in GitHub Desktop.
import config
import torch
import torch.optim as optim
from model import YOLOv3
from tqdm import tqdm
from utils import (
mean_average_precision,
cells_to_bboxes,
get_evaluation_bboxes,
save_checkpoint,
load_checkpoint,
check_class_accuracy,
get_loaders,
plot_couple_examples
)
from loss import YoloLoss
torch.backends.cudnn.benchmark = True
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment