YOLOv6代码解读[12] yolov6/models/yolo.py文件解读

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import math
import torch
import torch.nn as nn
import torch.nn.functional as F
from yolov6.layers.common import *
root@autodl-container-74204da04f-a96714b5:~/autodl-tmp/yolov8_ws# /root/miniconda3/bin/python /root/autodl-tmp/yolov8_ws/src/ultralytics/scripts/train/coal_mine_train.py 开始YOLOv8训练... 数据集: /root/autodl-tmp/yolov8_ws/src/ultralytics/dataset/hydraulic_support_guard_plate_data2023_yolo/dataset.yaml WARNING ⚠️ no model scale passed. Assuming scale='n'. Traceback (most recent call last): File "/root/autodl-tmp/yolov8_ws/src/ultralytics/scripts/train/coal_mine_train.py", line 29, in <module> main() File "/root/autodl-tmp/yolov8_ws/src/ultralytics/scripts/train/coal_mine_train.py", line 9, in main model = YOLO('/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/cfg/models/v8/yolov8-cmh.yaml') File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/models/yolo/model.py", line 79, in __init__ super().__init__(model=model, task=task, verbose=verbose) File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/engine/model.py", line 149, in __init__ self._new(model, task=task, verbose=verbose) File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/engine/model.py", line 261, in _new self.model = (model or self._smart_load("model"))(cfg_dict, verbose=verbose and RANK == -1) # build model File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/nn/tasks.py", line 435, in __init__ m.stride = torch.tensor([s / x.shape[-2] for x in _forward(torch.zeros(1, ch, s, s))]) # forward File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/nn/tasks.py", line 431, in _forward return self.forward(x)[0] if isinstance(m, (Segment, YOLOESegment, Pose, OBB)) else self.forward(x) File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/nn/tasks.py", line 157, in forward return self.predict(x, *args, **kwargs) File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ultralytics/nn/tasks.py", line 175, in predict return self._predict_once(x, profile, visualize, embed) File "/root/autodl-tmp/yolov8_ws/src/ultralytics/ul
07-24
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

我是天才很好

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值