Unverified 提交 a12698f4 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update ci-testing.yml reflecting sys.path additions (#869)

* Update ci-testing.yml * Update yolo.py * Update ci-testing.yml * Update export.py * Update yolo.py
上级 d55c326d
...@@ -53,13 +53,13 @@ jobs: ...@@ -53,13 +53,13 @@ jobs:
- name: Download data - name: Download data
run: | run: |
curl -L -o temp.zip https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip # curl -L -o tmp.zip https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
unzip -q temp.zip -d ../ # unzip -q tmp.zip -d ../
rm temp.zip # rm tmp.zip
- name: Tests workflow - name: Tests workflow
run: | run: |
export PYTHONPATH="$PWD" # to run *.py. files in subdirectories # export PYTHONPATH="$PWD" # to run '$ python *.py' files in subdirectories
di=cpu # inference devices # define device di=cpu # inference devices # define device
# train # train
......
...@@ -5,13 +5,13 @@ Usage: ...@@ -5,13 +5,13 @@ Usage:
""" """
import argparse import argparse
import sys
sys.path.append('./') # to run '$ python *.py' files in subdirectories
import torch import torch
import torch.nn as nn import torch.nn as nn
import sys
sys.path.append('./')
import models import models
from models.experimental import attempt_load from models.experimental import attempt_load
from utils.activations import Hardswish from utils.activations import Hardswish
......
import argparse import argparse
import logging import logging
import math import math
import sys
from copy import deepcopy from copy import deepcopy
from pathlib import Path from pathlib import Path
sys.path.append('./') # to run '$ python *.py' files in subdirectories
logger = logging.getLogger(__name__)
import torch import torch
import torch.nn as nn import torch.nn as nn
...@@ -13,8 +17,6 @@ from utils.general import check_anchor_order, make_divisible, check_file, set_lo ...@@ -13,8 +17,6 @@ from utils.general import check_anchor_order, make_divisible, check_file, set_lo
from utils.torch_utils import ( from utils.torch_utils import (
time_synchronized, fuse_conv_and_bn, model_info, scale_img, initialize_weights, select_device) time_synchronized, fuse_conv_and_bn, model_info, scale_img, initialize_weights, select_device)
logger = logging.getLogger(__name__)
class Detect(nn.Module): class Detect(nn.Module):
stride = None # strides computed during build stride = None # strides computed during build
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论