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
pull/1086/head
Glenn Jocher 2020-10-04 18:50:32 +02:00 committed by GitHub
parent d55c326dda
commit a12698f4c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 11 additions and 9 deletions

View File

@ -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

View File

@ -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

View File

@ -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