ABAO77 commited on
Commit
fd4f922
·
1 Parent(s): f113e60

print path

Browse files
__pycache__/a_control_classification.cpython-311.pyc ADDED
Binary file (1.46 kB). View file
 
__pycache__/a_utils_func_2_model.cpython-311.pyc ADDED
Binary file (8.89 kB). View file
 
__pycache__/setting_AI.cpython-311.pyc ADDED
Binary file (679 Bytes). View file
 
classification/__pycache__/__init__.cpython-311.pyc CHANGED
Binary files a/classification/__pycache__/__init__.cpython-311.pyc and b/classification/__pycache__/__init__.cpython-311.pyc differ
 
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
@@ -44,6 +44,7 @@ def load_model(model_path: str):
44
 
45
  dirname = os.path.dirname(__file__)
46
  session = load_model(os.path.join(dirname, "./model/model_16.onnx"))
 
47
 
48
 
49
  def prepare_input(image: NDArray[np.uint8]) -> NDArray[np.float16]:
@@ -179,7 +180,6 @@ def process_video(
179
  def inference_classification(image):
180
 
181
  predicted_class, probabilities = inference(image)
182
- print(f"Predicted Class: {predicted_class}, Probabilities: {probabilities}")
183
  ADD_DATA_CSV_CLASSIFICATION(predicted_class)
184
  CHECK_CSV_CLASSIFICATION()
185
 
 
44
 
45
  dirname = os.path.dirname(__file__)
46
  session = load_model(os.path.join(dirname, "./model/model_16.onnx"))
47
+ print("model_path classification", os.path.join(dirname, "./model/model_16.onnx"))
48
 
49
 
50
  def prepare_input(image: NDArray[np.uint8]) -> NDArray[np.float16]:
 
180
  def inference_classification(image):
181
 
182
  predicted_class, probabilities = inference(image)
 
183
  ADD_DATA_CSV_CLASSIFICATION(predicted_class)
184
  CHECK_CSV_CLASSIFICATION()
185
 
dataCSV/classification.csv CHANGED
@@ -246,1490 +246,3 @@ RIGHT
246
  RIGHT
247
  RIGHT
248
  RIGHT
249
- RIGHT
250
- RIGHT
251
- RIGHT
252
- RIGHT
253
- RIGHT
254
- RIGHT
255
- RIGHT
256
- RIGHT
257
- RIGHT
258
- RIGHT
259
- RIGHT
260
- RIGHT
261
- RIGHT
262
- RIGHT
263
- RIGHT
264
- RIGHT
265
- RIGHT
266
- RIGHT
267
- RIGHT
268
- RIGHT
269
- RIGHT
270
- RIGHT
271
- STRAIGHT
272
- LEFT
273
- RIGHT
274
- RIGHT
275
- RIGHT
276
- RIGHT
277
- RIGHT
278
- RIGHT
279
- RIGHT
280
- RIGHT
281
- LEFT
282
- LEFT
283
- LEFT
284
- LEFT
285
- RIGHT
286
- RIGHT
287
- RIGHT
288
- RIGHT
289
- RIGHT
290
- RIGHT
291
- RIGHT
292
- RIGHT
293
- RIGHT
294
- RIGHT
295
- RIGHT
296
- RIGHT
297
- RIGHT
298
- RIGHT
299
- RIGHT
300
- RIGHT
301
- RIGHT
302
- RIGHT
303
- RIGHT
304
- RIGHT
305
- RIGHT
306
- RIGHT
307
- RIGHT
308
- RIGHT
309
- RIGHT
310
- RIGHT
311
- RIGHT
312
- RIGHT
313
- RIGHT
314
- RIGHT
315
- RIGHT
316
- RIGHT
317
- RIGHT
318
- RIGHT
319
- RIGHT
320
- RIGHT
321
- RIGHT
322
- RIGHT
323
- RIGHT
324
- RIGHT
325
- RIGHT
326
- RIGHT
327
- RIGHT
328
- RIGHT
329
- RIGHT
330
- RIGHT
331
- RIGHT
332
- RIGHT
333
- RIGHT
334
- RIGHT
335
- RIGHT
336
- RIGHT
337
- RIGHT
338
- RIGHT
339
- RIGHT
340
- RIGHT
341
- RIGHT
342
- RIGHT
343
- RIGHT
344
- RIGHT
345
- RIGHT
346
- RIGHT
347
- RIGHT
348
- RIGHT
349
- STRAIGHT
350
- STRAIGHT
351
- STRAIGHT
352
- STRAIGHT
353
- STRAIGHT
354
- STRAIGHT
355
- STRAIGHT
356
- STRAIGHT
357
- STRAIGHT
358
- STRAIGHT
359
- RIGHT
360
- STRAIGHT
361
- STRAIGHT
362
- STRAIGHT
363
- STRAIGHT
364
- STRAIGHT
365
- RIGHT
366
- RIGHT
367
- RIGHT
368
- RIGHT
369
- RIGHT
370
- RIGHT
371
- RIGHT
372
- RIGHT
373
- RIGHT
374
- RIGHT
375
- RIGHT
376
- LEFT
377
- LEFT
378
- RIGHT
379
- RIGHT
380
- RIGHT
381
- RIGHT
382
- RIGHT
383
- RIGHT
384
- RIGHT
385
- RIGHT
386
- STRAIGHT
387
- STRAIGHT
388
- STRAIGHT
389
- RIGHT
390
- RIGHT
391
- RIGHT
392
- RIGHT
393
- RIGHT
394
- RIGHT
395
- RIGHT
396
- STRAIGHT
397
- RIGHT
398
- RIGHT
399
- RIGHT
400
- RIGHT
401
- RIGHT
402
- RIGHT
403
- RIGHT
404
- RIGHT
405
- RIGHT
406
- RIGHT
407
- RIGHT
408
- RIGHT
409
- RIGHT
410
- RIGHT
411
- RIGHT
412
- RIGHT
413
- RIGHT
414
- RIGHT
415
- RIGHT
416
- RIGHT
417
- RIGHT
418
- RIGHT
419
- RIGHT
420
- RIGHT
421
- RIGHT
422
- RIGHT
423
- RIGHT
424
- RIGHT
425
- RIGHT
426
- RIGHT
427
- RIGHT
428
- RIGHT
429
- RIGHT
430
- RIGHT
431
- RIGHT
432
- RIGHT
433
- RIGHT
434
- RIGHT
435
- RIGHT
436
- RIGHT
437
- RIGHT
438
- RIGHT
439
- RIGHT
440
- RIGHT
441
- RIGHT
442
- RIGHT
443
- RIGHT
444
- RIGHT
445
- STRAIGHT
446
- RIGHT
447
- RIGHT
448
- STRAIGHT
449
- RIGHT
450
- RIGHT
451
- RIGHT
452
- RIGHT
453
- STRAIGHT
454
- STRAIGHT
455
- STRAIGHT
456
- LEFT
457
- STRAIGHT
458
- STRAIGHT
459
- STRAIGHT
460
- STRAIGHT
461
- RIGHT
462
- RIGHT
463
- STRAIGHT
464
- STRAIGHT
465
- STRAIGHT
466
- RIGHT
467
- RIGHT
468
- RIGHT
469
- LEFT
470
- STRAIGHT
471
- LEFT
472
- RIGHT
473
- RIGHT
474
- RIGHT
475
- RIGHT
476
- RIGHT
477
- RIGHT
478
- RIGHT
479
- RIGHT
480
- RIGHT
481
- RIGHT
482
- RIGHT
483
- RIGHT
484
- RIGHT
485
- RIGHT
486
- RIGHT
487
- RIGHT
488
- RIGHT
489
- RIGHT
490
- RIGHT
491
- RIGHT
492
- RIGHT
493
- RIGHT
494
- RIGHT
495
- RIGHT
496
- RIGHT
497
- RIGHT
498
- RIGHT
499
- RIGHT
500
- RIGHT
501
- RIGHT
502
- RIGHT
503
- RIGHT
504
- RIGHT
505
- RIGHT
506
- RIGHT
507
- RIGHT
508
- RIGHT
509
- RIGHT
510
- RIGHT
511
- RIGHT
512
- STRAIGHT
513
- STRAIGHT
514
- STRAIGHT
515
- STRAIGHT
516
- STRAIGHT
517
- STRAIGHT
518
- STRAIGHT
519
- STRAIGHT
520
- STRAIGHT
521
- RIGHT
522
- RIGHT
523
- RIGHT
524
- STRAIGHT
525
- STRAIGHT
526
- STRAIGHT
527
- STRAIGHT
528
- STRAIGHT
529
- STRAIGHT
530
- STRAIGHT
531
- STRAIGHT
532
- STRAIGHT
533
- STRAIGHT
534
- STRAIGHT
535
- STRAIGHT
536
- LEFT
537
- LEFT
538
- LEFT
539
- LEFT
540
- LEFT
541
- LEFT
542
- LEFT
543
- LEFT
544
- LEFT
545
- LEFT
546
- LEFT
547
- LEFT
548
- RIGHT
549
- RIGHT
550
- RIGHT
551
- RIGHT
552
- RIGHT
553
- RIGHT
554
- RIGHT
555
- RIGHT
556
- RIGHT
557
- RIGHT
558
- RIGHT
559
- RIGHT
560
- RIGHT
561
- RIGHT
562
- RIGHT
563
- RIGHT
564
- RIGHT
565
- RIGHT
566
- RIGHT
567
- RIGHT
568
- RIGHT
569
- RIGHT
570
- RIGHT
571
- RIGHT
572
- RIGHT
573
- RIGHT
574
- RIGHT
575
- RIGHT
576
- RIGHT
577
- RIGHT
578
- RIGHT
579
- RIGHT
580
- RIGHT
581
- RIGHT
582
- RIGHT
583
- RIGHT
584
- RIGHT
585
- RIGHT
586
- RIGHT
587
- RIGHT
588
- RIGHT
589
- RIGHT
590
- RIGHT
591
- RIGHT
592
- RIGHT
593
- RIGHT
594
- RIGHT
595
- RIGHT
596
- RIGHT
597
- RIGHT
598
- RIGHT
599
- RIGHT
600
- RIGHT
601
- RIGHT
602
- RIGHT
603
- RIGHT
604
- RIGHT
605
- RIGHT
606
- RIGHT
607
- RIGHT
608
- RIGHT
609
- RIGHT
610
- RIGHT
611
- RIGHT
612
- RIGHT
613
- RIGHT
614
- RIGHT
615
- RIGHT
616
- RIGHT
617
- RIGHT
618
- RIGHT
619
- RIGHT
620
- RIGHT
621
- RIGHT
622
- RIGHT
623
- RIGHT
624
- RIGHT
625
- RIGHT
626
- RIGHT
627
- RIGHT
628
- RIGHT
629
- RIGHT
630
- RIGHT
631
- RIGHT
632
- RIGHT
633
- RIGHT
634
- RIGHT
635
- RIGHT
636
- RIGHT
637
- RIGHT
638
- RIGHT
639
- RIGHT
640
- RIGHT
641
- RIGHT
642
- RIGHT
643
- RIGHT
644
- RIGHT
645
- RIGHT
646
- RIGHT
647
- RIGHT
648
- RIGHT
649
- RIGHT
650
- RIGHT
651
- RIGHT
652
- RIGHT
653
- RIGHT
654
- RIGHT
655
- RIGHT
656
- RIGHT
657
- RIGHT
658
- RIGHT
659
- RIGHT
660
- RIGHT
661
- RIGHT
662
- RIGHT
663
- RIGHT
664
- RIGHT
665
- RIGHT
666
- RIGHT
667
- RIGHT
668
- RIGHT
669
- RIGHT
670
- RIGHT
671
- RIGHT
672
- RIGHT
673
- RIGHT
674
- RIGHT
675
- RIGHT
676
- RIGHT
677
- RIGHT
678
- RIGHT
679
- RIGHT
680
- RIGHT
681
- RIGHT
682
- RIGHT
683
- RIGHT
684
- RIGHT
685
- RIGHT
686
- RIGHT
687
- RIGHT
688
- RIGHT
689
- RIGHT
690
- RIGHT
691
- RIGHT
692
- RIGHT
693
- RIGHT
694
- RIGHT
695
- RIGHT
696
- RIGHT
697
- RIGHT
698
- RIGHT
699
- RIGHT
700
- RIGHT
701
- RIGHT
702
- RIGHT
703
- RIGHT
704
- RIGHT
705
- RIGHT
706
- RIGHT
707
- RIGHT
708
- RIGHT
709
- RIGHT
710
- RIGHT
711
- STRAIGHT
712
- STRAIGHT
713
- STRAIGHT
714
- STRAIGHT
715
- STRAIGHT
716
- STRAIGHT
717
- STRAIGHT
718
- STRAIGHT
719
- RIGHT
720
- RIGHT
721
- RIGHT
722
- RIGHT
723
- RIGHT
724
- RIGHT
725
- RIGHT
726
- RIGHT
727
- RIGHT
728
- RIGHT
729
- RIGHT
730
- RIGHT
731
- RIGHT
732
- RIGHT
733
- RIGHT
734
- RIGHT
735
- RIGHT
736
- RIGHT
737
- RIGHT
738
- RIGHT
739
- RIGHT
740
- LEFT
741
- LEFT
742
- LEFT
743
- STRAIGHT
744
- RIGHT
745
- RIGHT
746
- RIGHT
747
- RIGHT
748
- RIGHT
749
- RIGHT
750
- RIGHT
751
- RIGHT
752
- RIGHT
753
- RIGHT
754
- RIGHT
755
- RIGHT
756
- LEFT
757
- RIGHT
758
- RIGHT
759
- RIGHT
760
- RIGHT
761
- LEFT
762
- LEFT
763
- LEFT
764
- LEFT
765
- LEFT
766
- LEFT
767
- LEFT
768
- RIGHT
769
- LEFT
770
- LEFT
771
- RIGHT
772
- RIGHT
773
- RIGHT
774
- RIGHT
775
- RIGHT
776
- RIGHT
777
- RIGHT
778
- RIGHT
779
- RIGHT
780
- RIGHT
781
- RIGHT
782
- RIGHT
783
- RIGHT
784
- RIGHT
785
- STRAIGHT
786
- STRAIGHT
787
- STRAIGHT
788
- STRAIGHT
789
- STRAIGHT
790
- STRAIGHT
791
- STRAIGHT
792
- STRAIGHT
793
- STRAIGHT
794
- STRAIGHT
795
- STRAIGHT
796
- RIGHT
797
- RIGHT
798
- RIGHT
799
- RIGHT
800
- RIGHT
801
- RIGHT
802
- RIGHT
803
- RIGHT
804
- RIGHT
805
- RIGHT
806
- RIGHT
807
- STRAIGHT
808
- STRAIGHT
809
- STRAIGHT
810
- RIGHT
811
- STRAIGHT
812
- STRAIGHT
813
- RIGHT
814
- STRAIGHT
815
- STRAIGHT
816
- STRAIGHT
817
- RIGHT
818
- RIGHT
819
- STRAIGHT
820
- STRAIGHT
821
- RIGHT
822
- RIGHT
823
- RIGHT
824
- RIGHT
825
- RIGHT
826
- RIGHT
827
- RIGHT
828
- STRAIGHT
829
- STRAIGHT
830
- STRAIGHT
831
- STRAIGHT
832
- STRAIGHT
833
- STRAIGHT
834
- STRAIGHT
835
- STRAIGHT
836
- STRAIGHT
837
- STRAIGHT
838
- STRAIGHT
839
- STRAIGHT
840
- STRAIGHT
841
- STRAIGHT
842
- STRAIGHT
843
- STRAIGHT
844
- STRAIGHT
845
- RIGHT
846
- RIGHT
847
- RIGHT
848
- RIGHT
849
- RIGHT
850
- RIGHT
851
- RIGHT
852
- RIGHT
853
- RIGHT
854
- RIGHT
855
- RIGHT
856
- RIGHT
857
- RIGHT
858
- RIGHT
859
- RIGHT
860
- RIGHT
861
- RIGHT
862
- RIGHT
863
- RIGHT
864
- RIGHT
865
- RIGHT
866
- RIGHT
867
- RIGHT
868
- RIGHT
869
- RIGHT
870
- RIGHT
871
- RIGHT
872
- RIGHT
873
- RIGHT
874
- RIGHT
875
- RIGHT
876
- RIGHT
877
- RIGHT
878
- RIGHT
879
- RIGHT
880
- RIGHT
881
- RIGHT
882
- RIGHT
883
- RIGHT
884
- RIGHT
885
- RIGHT
886
- RIGHT
887
- RIGHT
888
- RIGHT
889
- RIGHT
890
- RIGHT
891
- RIGHT
892
- RIGHT
893
- RIGHT
894
- RIGHT
895
- RIGHT
896
- RIGHT
897
- RIGHT
898
- RIGHT
899
- RIGHT
900
- RIGHT
901
- RIGHT
902
- RIGHT
903
- RIGHT
904
- RIGHT
905
- RIGHT
906
- RIGHT
907
- RIGHT
908
- RIGHT
909
- RIGHT
910
- RIGHT
911
- RIGHT
912
- RIGHT
913
- RIGHT
914
- RIGHT
915
- RIGHT
916
- RIGHT
917
- RIGHT
918
- RIGHT
919
- RIGHT
920
- RIGHT
921
- RIGHT
922
- RIGHT
923
- RIGHT
924
- RIGHT
925
- RIGHT
926
- RIGHT
927
- RIGHT
928
- STRAIGHT
929
- STRAIGHT
930
- RIGHT
931
- RIGHT
932
- RIGHT
933
- RIGHT
934
- RIGHT
935
- RIGHT
936
- RIGHT
937
- RIGHT
938
- RIGHT
939
- RIGHT
940
- RIGHT
941
- RIGHT
942
- RIGHT
943
- RIGHT
944
- RIGHT
945
- RIGHT
946
- RIGHT
947
- RIGHT
948
- RIGHT
949
- RIGHT
950
- RIGHT
951
- RIGHT
952
- RIGHT
953
- RIGHT
954
- RIGHT
955
- RIGHT
956
- RIGHT
957
- RIGHT
958
- RIGHT
959
- RIGHT
960
- STRAIGHT
961
- STRAIGHT
962
- STRAIGHT
963
- STRAIGHT
964
- STRAIGHT
965
- RIGHT
966
- STRAIGHT
967
- STRAIGHT
968
- STRAIGHT
969
- STRAIGHT
970
- STRAIGHT
971
- STRAIGHT
972
- STRAIGHT
973
- STRAIGHT
974
- STRAIGHT
975
- STRAIGHT
976
- STRAIGHT
977
- STRAIGHT
978
- STRAIGHT
979
- STRAIGHT
980
- STRAIGHT
981
- STRAIGHT
982
- STRAIGHT
983
- STRAIGHT
984
- STRAIGHT
985
- STRAIGHT
986
- STRAIGHT
987
- STRAIGHT
988
- STRAIGHT
989
- STRAIGHT
990
- STRAIGHT
991
- STRAIGHT
992
- STRAIGHT
993
- STRAIGHT
994
- STRAIGHT
995
- STRAIGHT
996
- STRAIGHT
997
- STRAIGHT
998
- STRAIGHT
999
- STRAIGHT
1000
- STRAIGHT
1001
- STRAIGHT
1002
- STRAIGHT
1003
- STRAIGHT
1004
- LEFT
1005
- LEFT
1006
- LEFT
1007
- LEFT
1008
- LEFT
1009
- STRAIGHT
1010
- STRAIGHT
1011
- STRAIGHT
1012
- STRAIGHT
1013
- LEFT
1014
- STRAIGHT
1015
- STRAIGHT
1016
- STRAIGHT
1017
- STRAIGHT
1018
- STRAIGHT
1019
- STRAIGHT
1020
- STRAIGHT
1021
- STRAIGHT
1022
- STRAIGHT
1023
- STRAIGHT
1024
- STRAIGHT
1025
- STRAIGHT
1026
- STRAIGHT
1027
- STRAIGHT
1028
- STRAIGHT
1029
- STRAIGHT
1030
- STRAIGHT
1031
- STRAIGHT
1032
- LEFT
1033
- STRAIGHT
1034
- LEFT
1035
- LEFT
1036
- LEFT
1037
- LEFT
1038
- LEFT
1039
- LEFT
1040
- LEFT
1041
- LEFT
1042
- LEFT
1043
- LEFT
1044
- LEFT
1045
- LEFT
1046
- LEFT
1047
- LEFT
1048
- LEFT
1049
- LEFT
1050
- LEFT
1051
- LEFT
1052
- LEFT
1053
- LEFT
1054
- LEFT
1055
- LEFT
1056
- LEFT
1057
- LEFT
1058
- LEFT
1059
- LEFT
1060
- LEFT
1061
- LEFT
1062
- LEFT
1063
- LEFT
1064
- LEFT
1065
- LEFT
1066
- LEFT
1067
- LEFT
1068
- LEFT
1069
- LEFT
1070
- LEFT
1071
- LEFT
1072
- LEFT
1073
- LEFT
1074
- LEFT
1075
- LEFT
1076
- LEFT
1077
- LEFT
1078
- LEFT
1079
- LEFT
1080
- LEFT
1081
- LEFT
1082
- LEFT
1083
- LEFT
1084
- LEFT
1085
- LEFT
1086
- LEFT
1087
- LEFT
1088
- LEFT
1089
- LEFT
1090
- LEFT
1091
- LEFT
1092
- LEFT
1093
- LEFT
1094
- LEFT
1095
- LEFT
1096
- LEFT
1097
- LEFT
1098
- LEFT
1099
- LEFT
1100
- LEFT
1101
- LEFT
1102
- LEFT
1103
- LEFT
1104
- LEFT
1105
- LEFT
1106
- LEFT
1107
- LEFT
1108
- LEFT
1109
- RIGHT
1110
- RIGHT
1111
- RIGHT
1112
- RIGHT
1113
- RIGHT
1114
- RIGHT
1115
- RIGHT
1116
- RIGHT
1117
- RIGHT
1118
- RIGHT
1119
- RIGHT
1120
- RIGHT
1121
- RIGHT
1122
- RIGHT
1123
- RIGHT
1124
- LEFT
1125
- LEFT
1126
- LEFT
1127
- LEFT
1128
- LEFT
1129
- LEFT
1130
- LEFT
1131
- LEFT
1132
- LEFT
1133
- LEFT
1134
- LEFT
1135
- LEFT
1136
- LEFT
1137
- LEFT
1138
- LEFT
1139
- LEFT
1140
- LEFT
1141
- LEFT
1142
- LEFT
1143
- LEFT
1144
- LEFT
1145
- LEFT
1146
- LEFT
1147
- LEFT
1148
- LEFT
1149
- LEFT
1150
- LEFT
1151
- LEFT
1152
- LEFT
1153
- LEFT
1154
- LEFT
1155
- LEFT
1156
- LEFT
1157
- LEFT
1158
- LEFT
1159
- LEFT
1160
- RIGHT
1161
- RIGHT
1162
- RIGHT
1163
- RIGHT
1164
- RIGHT
1165
- RIGHT
1166
- RIGHT
1167
- RIGHT
1168
- RIGHT
1169
- RIGHT
1170
- RIGHT
1171
- RIGHT
1172
- RIGHT
1173
- RIGHT
1174
- RIGHT
1175
- RIGHT
1176
- RIGHT
1177
- RIGHT
1178
- STRAIGHT
1179
- RIGHT
1180
- RIGHT
1181
- RIGHT
1182
- RIGHT
1183
- RIGHT
1184
- RIGHT
1185
- LEFT
1186
- LEFT
1187
- LEFT
1188
- RIGHT
1189
- RIGHT
1190
- LEFT
1191
- RIGHT
1192
- RIGHT
1193
- RIGHT
1194
- RIGHT
1195
- RIGHT
1196
- RIGHT
1197
- RIGHT
1198
- RIGHT
1199
- RIGHT
1200
- STRAIGHT
1201
- STRAIGHT
1202
- STRAIGHT
1203
- STRAIGHT
1204
- STRAIGHT
1205
- STRAIGHT
1206
- STRAIGHT
1207
- STRAIGHT
1208
- STRAIGHT
1209
- STRAIGHT
1210
- STRAIGHT
1211
- LEFT
1212
- STRAIGHT
1213
- STRAIGHT
1214
- STRAIGHT
1215
- STRAIGHT
1216
- RIGHT
1217
- STRAIGHT
1218
- RIGHT
1219
- RIGHT
1220
- STRAIGHT
1221
- STRAIGHT
1222
- STRAIGHT
1223
- STRAIGHT
1224
- STRAIGHT
1225
- RIGHT
1226
- STRAIGHT
1227
- STRAIGHT
1228
- STRAIGHT
1229
- STRAIGHT
1230
- STRAIGHT
1231
- STRAIGHT
1232
- STRAIGHT
1233
- STRAIGHT
1234
- RIGHT
1235
- STRAIGHT
1236
- STRAIGHT
1237
- STRAIGHT
1238
- STRAIGHT
1239
- STRAIGHT
1240
- STRAIGHT
1241
- STRAIGHT
1242
- STRAIGHT
1243
- STRAIGHT
1244
- STRAIGHT
1245
- STRAIGHT
1246
- STRAIGHT
1247
- STRAIGHT
1248
- STRAIGHT
1249
- STRAIGHT
1250
- STRAIGHT
1251
- STRAIGHT
1252
- STRAIGHT
1253
- STRAIGHT
1254
- STRAIGHT
1255
- STRAIGHT
1256
- STRAIGHT
1257
- STRAIGHT
1258
- STRAIGHT
1259
- STRAIGHT
1260
- STRAIGHT
1261
- STRAIGHT
1262
- STRAIGHT
1263
- STRAIGHT
1264
- STRAIGHT
1265
- STRAIGHT
1266
- STRAIGHT
1267
- STRAIGHT
1268
- STRAIGHT
1269
- STRAIGHT
1270
- STRAIGHT
1271
- STRAIGHT
1272
- STRAIGHT
1273
- STRAIGHT
1274
- STRAIGHT
1275
- STRAIGHT
1276
- STRAIGHT
1277
- STRAIGHT
1278
- STRAIGHT
1279
- STRAIGHT
1280
- STRAIGHT
1281
- STRAIGHT
1282
- STRAIGHT
1283
- STRAIGHT
1284
- STRAIGHT
1285
- STRAIGHT
1286
- STRAIGHT
1287
- STRAIGHT
1288
- STRAIGHT
1289
- STRAIGHT
1290
- STRAIGHT
1291
- STRAIGHT
1292
- STRAIGHT
1293
- STRAIGHT
1294
- STRAIGHT
1295
- STRAIGHT
1296
- RIGHT
1297
- STRAIGHT
1298
- STRAIGHT
1299
- RIGHT
1300
- STRAIGHT
1301
- STRAIGHT
1302
- STRAIGHT
1303
- STRAIGHT
1304
- STRAIGHT
1305
- STRAIGHT
1306
- STRAIGHT
1307
- STRAIGHT
1308
- STRAIGHT
1309
- STRAIGHT
1310
- STRAIGHT
1311
- STRAIGHT
1312
- STRAIGHT
1313
- STRAIGHT
1314
- STRAIGHT
1315
- STRAIGHT
1316
- STRAIGHT
1317
- RIGHT
1318
- STRAIGHT
1319
- STRAIGHT
1320
- STRAIGHT
1321
- STRAIGHT
1322
- RIGHT
1323
- STRAIGHT
1324
- STRAIGHT
1325
- STRAIGHT
1326
- STRAIGHT
1327
- STRAIGHT
1328
- STRAIGHT
1329
- STRAIGHT
1330
- STRAIGHT
1331
- STRAIGHT
1332
- STRAIGHT
1333
- STRAIGHT
1334
- STRAIGHT
1335
- STRAIGHT
1336
- RIGHT
1337
- RIGHT
1338
- STRAIGHT
1339
- STRAIGHT
1340
- STRAIGHT
1341
- STRAIGHT
1342
- STRAIGHT
1343
- STRAIGHT
1344
- STRAIGHT
1345
- STRAIGHT
1346
- STRAIGHT
1347
- STRAIGHT
1348
- STRAIGHT
1349
- STRAIGHT
1350
- STRAIGHT
1351
- STRAIGHT
1352
- RIGHT
1353
- RIGHT
1354
- RIGHT
1355
- RIGHT
1356
- STRAIGHT
1357
- RIGHT
1358
- RIGHT
1359
- RIGHT
1360
- RIGHT
1361
- RIGHT
1362
- RIGHT
1363
- RIGHT
1364
- RIGHT
1365
- RIGHT
1366
- RIGHT
1367
- RIGHT
1368
- STRAIGHT
1369
- STRAIGHT
1370
- STRAIGHT
1371
- STRAIGHT
1372
- STRAIGHT
1373
- STRAIGHT
1374
- STRAIGHT
1375
- STRAIGHT
1376
- STRAIGHT
1377
- STRAIGHT
1378
- STRAIGHT
1379
- STRAIGHT
1380
- STRAIGHT
1381
- STRAIGHT
1382
- STRAIGHT
1383
- STRAIGHT
1384
- STRAIGHT
1385
- STRAIGHT
1386
- STRAIGHT
1387
- STRAIGHT
1388
- STRAIGHT
1389
- STRAIGHT
1390
- STRAIGHT
1391
- STRAIGHT
1392
- STRAIGHT
1393
- STRAIGHT
1394
- STRAIGHT
1395
- STRAIGHT
1396
- STRAIGHT
1397
- STRAIGHT
1398
- STRAIGHT
1399
- STRAIGHT
1400
- STRAIGHT
1401
- STRAIGHT
1402
- STRAIGHT
1403
- STRAIGHT
1404
- STRAIGHT
1405
- STRAIGHT
1406
- STRAIGHT
1407
- STRAIGHT
1408
- STRAIGHT
1409
- STRAIGHT
1410
- STRAIGHT
1411
- STRAIGHT
1412
- STRAIGHT
1413
- STRAIGHT
1414
- STRAIGHT
1415
- STRAIGHT
1416
- STRAIGHT
1417
- STRAIGHT
1418
- STRAIGHT
1419
- STRAIGHT
1420
- STRAIGHT
1421
- STRAIGHT
1422
- STRAIGHT
1423
- STRAIGHT
1424
- RIGHT
1425
- RIGHT
1426
- RIGHT
1427
- RIGHT
1428
- RIGHT
1429
- RIGHT
1430
- RIGHT
1431
- RIGHT
1432
- RIGHT
1433
- RIGHT
1434
- RIGHT
1435
- RIGHT
1436
- RIGHT
1437
- RIGHT
1438
- RIGHT
1439
- RIGHT
1440
- RIGHT
1441
- RIGHT
1442
- RIGHT
1443
- RIGHT
1444
- RIGHT
1445
- RIGHT
1446
- RIGHT
1447
- RIGHT
1448
- RIGHT
1449
- RIGHT
1450
- RIGHT
1451
- RIGHT
1452
- RIGHT
1453
- RIGHT
1454
- LEFT
1455
- RIGHT
1456
- RIGHT
1457
- RIGHT
1458
- RIGHT
1459
- RIGHT
1460
- RIGHT
1461
- RIGHT
1462
- LEFT
1463
- LEFT
1464
- LEFT
1465
- LEFT
1466
- LEFT
1467
- LEFT
1468
- LEFT
1469
- LEFT
1470
- LEFT
1471
- LEFT
1472
- RIGHT
1473
- RIGHT
1474
- RIGHT
1475
- RIGHT
1476
- RIGHT
1477
- RIGHT
1478
- RIGHT
1479
- RIGHT
1480
- RIGHT
1481
- RIGHT
1482
- RIGHT
1483
- LEFT
1484
- LEFT
1485
- LEFT
1486
- RIGHT
1487
- LEFT
1488
- RIGHT
1489
- LEFT
1490
- RIGHT
1491
- RIGHT
1492
- RIGHT
1493
- RIGHT
1494
- RIGHT
1495
- RIGHT
1496
- RIGHT
1497
- RIGHT
1498
- RIGHT
1499
- RIGHT
1500
- RIGHT
1501
- RIGHT
1502
- RIGHT
1503
- RIGHT
1504
- RIGHT
1505
- RIGHT
1506
- RIGHT
1507
- RIGHT
1508
- RIGHT
1509
- RIGHT
1510
- RIGHT
1511
- RIGHT
1512
- RIGHT
1513
- RIGHT
1514
- RIGHT
1515
- LEFT
1516
- LEFT
1517
- RIGHT
1518
- RIGHT
1519
- LEFT
1520
- LEFT
1521
- LEFT
1522
- LEFT
1523
- LEFT
1524
- LEFT
1525
- LEFT
1526
- RIGHT
1527
- RIGHT
1528
- RIGHT
1529
- RIGHT
1530
- RIGHT
1531
- STRAIGHT
1532
- STRAIGHT
1533
- RIGHT
1534
- RIGHT
1535
- RIGHT
1536
- RIGHT
1537
- RIGHT
1538
- STRAIGHT
1539
- RIGHT
1540
- RIGHT
1541
- STRAIGHT
1542
- STRAIGHT
1543
- RIGHT
1544
- RIGHT
1545
- RIGHT
1546
- STRAIGHT
1547
- RIGHT
1548
- STRAIGHT
1549
- RIGHT
1550
- RIGHT
1551
- RIGHT
1552
- RIGHT
1553
- RIGHT
1554
- RIGHT
1555
- RIGHT
1556
- RIGHT
1557
- RIGHT
1558
- RIGHT
1559
- STRAIGHT
1560
- STRAIGHT
1561
- RIGHT
1562
- STRAIGHT
1563
- STRAIGHT
1564
- STRAIGHT
1565
- RIGHT
1566
- STRAIGHT
1567
- STRAIGHT
1568
- STRAIGHT
1569
- RIGHT
1570
- RIGHT
1571
- STRAIGHT
1572
- RIGHT
1573
- RIGHT
1574
- RIGHT
1575
- LEFT
1576
- LEFT
1577
- LEFT
1578
- LEFT
1579
- LEFT
1580
- LEFT
1581
- LEFT
1582
- LEFT
1583
- LEFT
1584
- RIGHT
1585
- RIGHT
1586
- RIGHT
1587
- RIGHT
1588
- RIGHT
1589
- RIGHT
1590
- RIGHT
1591
- RIGHT
1592
- RIGHT
1593
- RIGHT
1594
- RIGHT
1595
- RIGHT
1596
- RIGHT
1597
- RIGHT
1598
- RIGHT
1599
- RIGHT
1600
- RIGHT
1601
- RIGHT
1602
- RIGHT
1603
- STRAIGHT
1604
- RIGHT
1605
- RIGHT
1606
- RIGHT
1607
- STRAIGHT
1608
- RIGHT
1609
- RIGHT
1610
- RIGHT
1611
- STRAIGHT
1612
- STRAIGHT
1613
- RIGHT
1614
- RIGHT
1615
- STRAIGHT
1616
- RIGHT
1617
- RIGHT
1618
- RIGHT
1619
- STRAIGHT
1620
- STRAIGHT
1621
- STRAIGHT
1622
- STRAIGHT
1623
- STRAIGHT
1624
- STRAIGHT
1625
- STRAIGHT
1626
- STRAIGHT
1627
- RIGHT
1628
- RIGHT
1629
- RIGHT
1630
- RIGHT
1631
- RIGHT
1632
- RIGHT
1633
- RIGHT
1634
- STRAIGHT
1635
- RIGHT
1636
- RIGHT
1637
- STRAIGHT
1638
- RIGHT
1639
- RIGHT
1640
- RIGHT
1641
- RIGHT
1642
- RIGHT
1643
- RIGHT
1644
- RIGHT
1645
- RIGHT
1646
- RIGHT
1647
- RIGHT
1648
- RIGHT
1649
- RIGHT
1650
- RIGHT
1651
- RIGHT
1652
- RIGHT
1653
- RIGHT
1654
- RIGHT
1655
- RIGHT
1656
- RIGHT
1657
- RIGHT
1658
- RIGHT
1659
- RIGHT
1660
- RIGHT
1661
- RIGHT
1662
- RIGHT
1663
- STRAIGHT
1664
- RIGHT
1665
- RIGHT
1666
- STRAIGHT
1667
- STRAIGHT
1668
- STRAIGHT
1669
- RIGHT
1670
- STRAIGHT
1671
- STRAIGHT
1672
- STRAIGHT
1673
- STRAIGHT
1674
- STRAIGHT
1675
- RIGHT
1676
- RIGHT
1677
- RIGHT
1678
- RIGHT
1679
- RIGHT
1680
- RIGHT
1681
- RIGHT
1682
- RIGHT
1683
- RIGHT
1684
- RIGHT
1685
- RIGHT
1686
- RIGHT
1687
- RIGHT
1688
- RIGHT
1689
- RIGHT
1690
- RIGHT
1691
- RIGHT
1692
- RIGHT
1693
- RIGHT
1694
- RIGHT
1695
- RIGHT
1696
- RIGHT
1697
- RIGHT
1698
- RIGHT
1699
- RIGHT
1700
- RIGHT
1701
- RIGHT
1702
- RIGHT
1703
- RIGHT
1704
- RIGHT
1705
- RIGHT
1706
- RIGHT
1707
- RIGHT
1708
- RIGHT
1709
- RIGHT
1710
- RIGHT
1711
- RIGHT
1712
- RIGHT
1713
- RIGHT
1714
- LEFT
1715
- LEFT
1716
- LEFT
1717
- LEFT
1718
- LEFT
1719
- LEFT
1720
- LEFT
1721
- LEFT
1722
- RIGHT
1723
- RIGHT
1724
- LEFT
1725
- LEFT
1726
- LEFT
1727
- LEFT
1728
- LEFT
1729
- LEFT
1730
- LEFT
1731
- LEFT
1732
- LEFT
1733
- RIGHT
1734
- RIGHT
1735
- RIGHT
 
246
  RIGHT
247
  RIGHT
248
  RIGHT
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
requirements.txt CHANGED
@@ -4,4 +4,5 @@ onnxruntime
4
  opencv-python
5
  serial
6
  pandas
7
- matplotlib
 
 
4
  opencv-python
5
  serial
6
  pandas
7
+ matplotlib
8
+ scipy
ultrafast/__pycache__/inference_onnx.cpython-311.pyc CHANGED
Binary files a/ultrafast/__pycache__/inference_onnx.cpython-311.pyc and b/ultrafast/__pycache__/inference_onnx.cpython-311.pyc differ
 
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/inference_onnx.py CHANGED
@@ -36,13 +36,10 @@ model_type = ModelType.TUSIMPLE
36
  dirname = os.path.dirname(__file__)
37
 
38
  model_path = os.path.join(dirname, "./models/tusimple_18_V1_fp16.onnx")
 
39
  lane_detector = UltrafastLaneDetector(model_path, model_type)
40
 
41
 
42
- def inference_detect_lane(image):
43
- return lane_detector.detect_lanes(image)
44
-
45
-
46
  tusimple_row_anchor = [
47
  64,
48
  68,
 
36
  dirname = os.path.dirname(__file__)
37
 
38
  model_path = os.path.join(dirname, "./models/tusimple_18_V1_fp16.onnx")
39
+ print("model_path", model_path)
40
  lane_detector = UltrafastLaneDetector(model_path, model_type)
41
 
42
 
 
 
 
 
43
  tusimple_row_anchor = [
44
  64,
45
  68,
ultrafast/ultrafastLaneDetector.py CHANGED
@@ -130,8 +130,9 @@ class UltrafastLaneDetector:
130
  self.initialize_model(model_path)
131
 
132
  def initialize_model(self, model_path):
 
133
 
134
- self.session = onnxruntime.InferenceSession(model_path)
135
 
136
  # Get model info
137
  self.getModel_input_details()
 
130
  self.initialize_model(model_path)
131
 
132
  def initialize_model(self, model_path):
133
+ providers = ["CPUExecutionProvider"]
134
 
135
+ self.session = onnxruntime.InferenceSession(model_path, providers=providers)
136
 
137
  # Get model info
138
  self.getModel_input_details()