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
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-10-04 18:50:32 +02:00
1 parent d55c326dda
commit a12698f4c9
3 files changed
+11 -9

No files matched your search

+4 -4
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
+3 -3
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
+4 -2
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