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.pypull/1086/head
parent
d55c326dda
commit
a12698f4c9
|
@ -53,13 +53,13 @@ jobs:
|
|||
|
||||
- name: Download data
|
||||
run: |
|
||||
curl -L -o temp.zip https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
|
||||
unzip -q temp.zip -d ../
|
||||
rm temp.zip
|
||||
# curl -L -o tmp.zip https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
|
||||
# unzip -q tmp.zip -d ../
|
||||
# rm tmp.zip
|
||||
|
||||
- name: Tests workflow
|
||||
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
|
||||
|
||||
# train
|
||||
|
|
|
@ -5,13 +5,13 @@ Usage:
|
|||
"""
|
||||
|
||||
import argparse
|
||||
import sys
|
||||
|
||||
sys.path.append('./') # to run '$ python *.py' files in subdirectories
|
||||
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
import sys
|
||||
sys.path.append('./')
|
||||
|
||||
import models
|
||||
from models.experimental import attempt_load
|
||||
from utils.activations import Hardswish
|
||||
|
|
|
@ -1,9 +1,13 @@
|
|||
import argparse
|
||||
import logging
|
||||
import math
|
||||
import sys
|
||||
from copy import deepcopy
|
||||
from pathlib import Path
|
||||
|
||||
sys.path.append('./') # to run '$ python *.py' files in subdirectories
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
|
@ -13,8 +17,6 @@ from utils.general import check_anchor_order, make_divisible, check_file, set_lo
|
|||
from utils.torch_utils import (
|
||||
time_synchronized, fuse_conv_and_bn, model_info, scale_img, initialize_weights, select_device)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class Detect(nn.Module):
|
||||
stride = None # strides computed during build
|
||||
|
|
Loading…
Reference in New Issue