直接看视频演示:
基于yolo的行人车辆雨伞安全帽卡车检测识别-完整代码数据环境-可直接运行_哔哩哔哩_bilibili
完整代码:
#-------------------------------------#
# 对数据集进行训练
#-------------------------------------#
import datetime
import os
import numpy as np
import torch
import torch.backends.cudnn as cudnn
import torch.distributed as dist
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import DataLoader
from nets.yolo import YoloBody
from nets.yolo_training import (ModelEMA, YOLOLoss, get_lr_scheduler,
set_optimizer_lr,