Spaces:
No application file
No application file
File size: 5,330 Bytes
f113e60 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 |
import csv
import pandas as pd
from statistics import mode
from setting_AI import *
from a_control_classification import USE_CLASSIFICATION
csv_path = "dataCSV/direction_control.csv"
csv_mask_path = "dataCSV/direction_control_mask.csv"
csv_straight_path = "dataCSV/direction_straight.csv"
csv_back_control_path = "dataCSV/back_control.csv"
csv_classification_path = "dataCSV/classification.csv"
def ADD_DATA_CSV_CLASSIFICATION(direction):
with open(csv_classification_path, mode='a', newline='', encoding='utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([direction])
data_csv = pd.read_csv(csv_classification_path)
if len(data_csv) == 10000:
file_start = pd.read_csv(csv_classification_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_classification_path, index=False)
def CLEAN_DATA_CSV_CLASSIFICATION():
file_start = pd.read_csv(csv_classification_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_classification_path, index=False)
file_start = pd.read_csv(csv_classification_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_classification_path, index=False)
def CHECK_CSV_CLASSIFICATION():
data_csv = pd.read_csv(csv_classification_path)
direction_list_to_mode = list(data_csv['direction'][-THRESHOLD_CLASSIFICATION:])
direction_mode = mode(direction_list_to_mode)
USE_CLASSIFICATION.change(direction_mode)
if direction_mode != USE_CLASSIFICATION.check():
CLEAN_DATA_CSV_CLASSIFICATION()
def ADD_DATA_CSV_MASK_DIRECTION(direction, angle):
with open(csv_mask_path, mode='a', newline='', encoding='utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([direction, angle])
data_csv = pd.read_csv(csv_mask_path)
if len(data_csv) == 10000:
file_start = pd.read_csv(csv_mask_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_mask_path, index=False)
def ADD_DATA_CSV_DIRECTION(direction, angle):
with open(csv_path, mode='a', newline='', encoding='utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([direction, angle])
def ADD_DATA_CSV_DIRECTION_STRAIGHT(direction, angle):
with open(csv_straight_path, mode='a', newline='', encoding='utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([direction, angle])
data_csv = pd.read_csv(csv_straight_path)
if len(data_csv) == 500:
CLEAN_DATA_CSV_DIRECTION_STRAIGHT()
def CLEAN_DATA_CSV_DIRECTION():
# Clear "direction_control.csv"
file_start = pd.read_csv(csv_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_path, index=False)
# Clear "direction_control_mask.csv"
file_start = pd.read_csv(csv_mask_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_mask_path, index=False)
def ADD_DATA_CSV_BACK_CONTROL(direction, angle):
with open(csv_back_control_path, mode='a', newline='', encoding='utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow([direction, angle])
def CLEAN_DATA_CSV_BACK_CONTROL():
# Clear "back_control.csv"
file_start = pd.read_csv(csv_back_control_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_back_control_path, index=False)
def CLEAN_DATA_CSV_DIRECTION_STRAIGHT():
# Clear "direction_control.csv"
file_start = pd.read_csv(csv_straight_path, nrows=0)
file_start_new = pd.DataFrame(columns=file_start.columns)
file_start_new.to_csv(csv_straight_path, index=False)
def BOTTOM_DATA_CSV_CHECK():
data_csv_ = pd.read_csv(csv_path)
last_row = data_csv_.iloc[-1]
return (last_row["direction"], last_row["angle"])
def CHECK_PUSH():
push_variable = None
dr_back, an_back = None, None
data_csv_ = pd.read_csv(csv_mask_path)
direction_list_to_mode = list(data_csv_['direction'][-count_control:])
if len(direction_list_to_mode) > 0:
direction_mode = mode(direction_list_to_mode)
max_angle = max(list(data_csv_['angle'][:count_control]))
if len(pd.read_csv(csv_path)) == 0:
dr_back, an_back = direction_mode, max_angle
ADD_DATA_CSV_DIRECTION(direction_mode, max_angle)
# ADD_DATA_CSV_BACK_CONTROL(direction_mode, max_angle)
return f"{direction_mode}:{max_angle:03d}", dr_back, an_back
else:
bottom_data_csv_check = BOTTOM_DATA_CSV_CHECK()
if bottom_data_csv_check[0] != direction_mode or (abs(bottom_data_csv_check[1] - max_angle) >= threshold_scale):
CLEAN_DATA_CSV_DIRECTION()
# ADD_DATA_CSV_DIRECTION(direction_mode, max_angle)
dr_back, an_back = direction_mode, max_angle
return f"{direction_mode}:{max_angle:03d}", dr_back, an_back
else:
return push_variable, dr_back, an_back
return push_variable, dr_back, an_back
|