Commit
·
691f3ae
1
Parent(s):
15a04e6
Fixing typos
Browse files- training.py +2 -5
training.py
CHANGED
|
@@ -8,15 +8,12 @@ from ultralytics import YOLO
|
|
| 8 |
|
| 9 |
try:
|
| 10 |
from synet.backends import get_backend
|
| 11 |
-
from synet.backends import get_backend
|
| 12 |
-
|
| 13 |
except ImportError:
|
| 14 |
import subprocess
|
| 15 |
print('Installing synet package')
|
| 16 |
subprocess.check_call([sys.executable, "-m", "pip", "install", "./synet_package[ultra]"])
|
| 17 |
# Import the file
|
| 18 |
from synet.backends import get_backend
|
| 19 |
-
from synet.backends import get_backend
|
| 20 |
|
| 21 |
backend = get_backend('ultralytics')
|
| 22 |
backend.patch()
|
|
@@ -33,8 +30,8 @@ class model_training:
|
|
| 33 |
|
| 34 |
self.model_select = 'detect'
|
| 35 |
self.model_options = {
|
| 36 |
-
'detect' : { cfg : 'synet_package/synet/zoo/ultralytics/sabre-detect-vga.yaml',
|
| 37 |
-
image_size: (480,640) }
|
| 38 |
}
|
| 39 |
|
| 40 |
|
|
|
|
| 8 |
|
| 9 |
try:
|
| 10 |
from synet.backends import get_backend
|
|
|
|
|
|
|
| 11 |
except ImportError:
|
| 12 |
import subprocess
|
| 13 |
print('Installing synet package')
|
| 14 |
subprocess.check_call([sys.executable, "-m", "pip", "install", "./synet_package[ultra]"])
|
| 15 |
# Import the file
|
| 16 |
from synet.backends import get_backend
|
|
|
|
| 17 |
|
| 18 |
backend = get_backend('ultralytics')
|
| 19 |
backend.patch()
|
|
|
|
| 30 |
|
| 31 |
self.model_select = 'detect'
|
| 32 |
self.model_options = {
|
| 33 |
+
'detect' : { 'cfg' : 'synet_package/synet/zoo/ultralytics/sabre-detect-vga.yaml',
|
| 34 |
+
'image_size' : (480,640) }
|
| 35 |
}
|
| 36 |
|
| 37 |
|