Hugging Face's logo

Spaces:
ABAO77
/
Autonomous_Car
No application file

App Files Community
ABAO77 commited on
Commit
50a5239
·
1 Parent(s): fc7a73b

tensort

Browse files
Files changed (5) hide show
  1. classification/__pycache__/inference_onnx.cpython-311.pyc +0 -0
  2. classification/inference_onnx.py +3 -1
  3. dataCSV/classification.csv +0 -206
  4. ultrafast/__pycache__/ultrafastLaneDetector.cpython-311.pyc +0 -0
  5. ultrafast/ultrafastLaneDetector.py +3 -1
classification/__pycache__/inference_onnx.cpython-311.pyc CHANGED
Binary files a/classification/__pycache__/inference_onnx.cpython-311.pyc and b/classification/__pycache__/inference_onnx.cpython-311.pyc differ
 
classification/inference_onnx.py CHANGED
@@ -36,7 +36,9 @@ def load_model(model_path: str):
36
  available_providers = ort.get_available_providers()
37
 
38
  if "CUDAExecutionProvider" in available_providers:
39
- providers = ["CUDAExecutionProvider", "CPUExecutionProvider"]
 
 
40
  print("Using CUDA provider")
41
  else:
42
  providers = ["CPUExecutionProvider"]
 
36
  available_providers = ort.get_available_providers()
37
 
38
  if "CUDAExecutionProvider" in available_providers:
39
+ # "TensorrtExecutionProvider"
40
+ # "CUDAExecutionProvider"
41
+ providers = ["TensorrtExecutionProvider", "CPUExecutionProvider"]
42
  print("Using CUDA provider")
43
  else:
44
  providers = ["CPUExecutionProvider"]
dataCSV/classification.csv CHANGED
@@ -40,209 +40,3 @@ LEFT
40
  LEFT
41
  LEFT
42
  LEFT
43
- LEFT
44
- LEFT
45
- LEFT
46
- LEFT
47
- RIGHT
48
- RIGHT
49
- RIGHT
50
- RIGHT
51
- RIGHT
52
- RIGHT
53
- RIGHT
54
- RIGHT
55
- RIGHT
56
- RIGHT
57
- RIGHT
58
- RIGHT
59
- RIGHT
60
- RIGHT
61
- RIGHT
62
- RIGHT
63
- RIGHT
64
- RIGHT
65
- RIGHT
66
- LEFT
67
- LEFT
68
- RIGHT
69
- RIGHT
70
- RIGHT
71
- RIGHT
72
- RIGHT
73
- RIGHT
74
- RIGHT
75
- RIGHT
76
- RIGHT
77
- RIGHT
78
- RIGHT
79
- RIGHT
80
- RIGHT
81
- RIGHT
82
- RIGHT
83
- RIGHT
84
- RIGHT
85
- RIGHT
86
- RIGHT
87
- RIGHT
88
- RIGHT
89
- RIGHT
90
- RIGHT
91
- RIGHT
92
- RIGHT
93
- RIGHT
94
- RIGHT
95
- RIGHT
96
- RIGHT
97
- RIGHT
98
- RIGHT
99
- RIGHT
100
- RIGHT
101
- RIGHT
102
- RIGHT
103
- RIGHT
104
- RIGHT
105
- RIGHT
106
- RIGHT
107
- RIGHT
108
- RIGHT
109
- RIGHT
110
- RIGHT
111
- RIGHT
112
- RIGHT
113
- RIGHT
114
- RIGHT
115
- RIGHT
116
- RIGHT
117
- RIGHT
118
- RIGHT
119
- RIGHT
120
- RIGHT
121
- RIGHT
122
- RIGHT
123
- RIGHT
124
- RIGHT
125
- RIGHT
126
- RIGHT
127
- RIGHT
128
- RIGHT
129
- RIGHT
130
- RIGHT
131
- RIGHT
132
- RIGHT
133
- RIGHT
134
- RIGHT
135
- RIGHT
136
- RIGHT
137
- RIGHT
138
- RIGHT
139
- RIGHT
140
- RIGHT
141
- RIGHT
142
- RIGHT
143
- RIGHT
144
- RIGHT
145
- RIGHT
146
- RIGHT
147
- RIGHT
148
- RIGHT
149
- RIGHT
150
- RIGHT
151
- RIGHT
152
- RIGHT
153
- RIGHT
154
- RIGHT
155
- RIGHT
156
- RIGHT
157
- RIGHT
158
- RIGHT
159
- RIGHT
160
- RIGHT
161
- RIGHT
162
- RIGHT
163
- RIGHT
164
- RIGHT
165
- RIGHT
166
- RIGHT
167
- RIGHT
168
- RIGHT
169
- RIGHT
170
- RIGHT
171
- RIGHT
172
- RIGHT
173
- RIGHT
174
- RIGHT
175
- RIGHT
176
- RIGHT
177
- RIGHT
178
- RIGHT
179
- RIGHT
180
- RIGHT
181
- RIGHT
182
- RIGHT
183
- RIGHT
184
- RIGHT
185
- RIGHT
186
- RIGHT
187
- RIGHT
188
- RIGHT
189
- RIGHT
190
- RIGHT
191
- RIGHT
192
- RIGHT
193
- RIGHT
194
- STRAIGHT
195
- STRAIGHT
196
- STRAIGHT
197
- STRAIGHT
198
- STRAIGHT
199
- STRAIGHT
200
- STRAIGHT
201
- STRAIGHT
202
- STRAIGHT
203
- STRAIGHT
204
- STRAIGHT
205
- STRAIGHT
206
- STRAIGHT
207
- STRAIGHT
208
- STRAIGHT
209
- STRAIGHT
210
- STRAIGHT
211
- LEFT
212
- STRAIGHT
213
- LEFT
214
- STRAIGHT
215
- LEFT
216
- RIGHT
217
- STRAIGHT
218
- RIGHT
219
- RIGHT
220
- RIGHT
221
- RIGHT
222
- RIGHT
223
- RIGHT
224
- RIGHT
225
- RIGHT
226
- RIGHT
227
- RIGHT
228
- RIGHT
229
- RIGHT
230
- RIGHT
231
- RIGHT
232
- RIGHT
233
- RIGHT
234
- RIGHT
235
- RIGHT
236
- RIGHT
237
- RIGHT
238
- RIGHT
239
- RIGHT
240
- RIGHT
241
- RIGHT
242
- RIGHT
243
- RIGHT
244
- RIGHT
245
- RIGHT
246
- RIGHT
247
- RIGHT
248
- RIGHT
 
40
  LEFT
41
  LEFT
42
  LEFT
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ultrafast/__pycache__/ultrafastLaneDetector.cpython-311.pyc CHANGED
Binary files a/ultrafast/__pycache__/ultrafastLaneDetector.cpython-311.pyc and b/ultrafast/__pycache__/ultrafastLaneDetector.cpython-311.pyc differ
 
ultrafast/ultrafastLaneDetector.py CHANGED
@@ -133,7 +133,9 @@ class UltrafastLaneDetector:
133
  available_providers = onnxruntime.get_available_providers()
134
 
135
  if 'CUDAExecutionProvider' in available_providers:
136
- providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
 
 
137
  print("Using CUDA provider")
138
  else:
139
  providers = ['CPUExecutionProvider']
 
133
  available_providers = onnxruntime.get_available_providers()
134
 
135
  if 'CUDAExecutionProvider' in available_providers:
136
+ # "TensorrtExecutionProvider"
137
+ # "CUDAExecutionProvider"
138
+ providers = ['TensorrtExecutionProvider', 'CPUExecutionProvider']
139
  print("Using CUDA provider")
140
  else:
141
  providers = ['CPUExecutionProvider']