CubeBit/StreetsignRecognition
1
0
Fork 0
mirror of https://github.com/3x4byte/StreetsignRecognition.git synced 2025-12-22 01:45:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

nupdated feature vectors | added functionality to specify number of training vectors

Browse Source
This commit is contained in:
Lukas K
2024-11-26 00:30:27 +01:00
parent 1cca9bfe4c
commit 2b1b183247
6 changed files with 17363 additions and 10416 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/all_data_splitter.py
View File

@@ -27,14 +27,14 @@ def split_all_data(all_data_path: str, future_training_path: str, future_testing
random.shuffle(training_data)
random.shuffle(testing_data)
with open(future_training_path, 'w') as csv_file:
writer = csv.writer(csv_file, delimiter=";")
writer.writerow(['filename','concept', 'vector'])
writer = csv.writer(csv_file, delimiter=";", lineterminator="\n")
writer.writerow(['brightness','filename','concept', 'vector'])
for sing in training_data:
writer.writerow(sing)
with open(future_testing_path, 'w') as csv_file:
writer = csv.writer(csv_file, delimiter=";")
writer.writerow(['filename','concept', 'vector'])
writer = csv.writer(csv_file, delimiter=";", lineterminator="\n")
writer.writerow(['brightness','filename','concept', 'vector'])
for sing in testing_data:
writer.writerow(sing)