mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

export.py folder sys.path addition (#1052)

pull/869/head
PetrDvoracek 2020-10-04 17:42:05 +02:00 committed by GitHub
parent 7220cee1d1
commit d55c326dda
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/export.py
View File

@ -9,6 +9,9 @@ import argparse
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