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
|
- 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:
|
||||||
"""
|
"""
|
||||||
|
|
||||||
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
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue