dgarrett-synaptics commited on
Commit
691f3ae
·
1 Parent(s): 15a04e6

Fixing typos

Browse files
Files changed (1) hide show
  1. 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