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

View File

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

View File

@ -1,9 +1,13 @@
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
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