2_train.py 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. """train"""
  2. from datetime import datetime
  3. import os
  4. import streamlit as st
  5. from db.models import subject
  6. from db.models import train
  7. from components.remove_style import hide_footer
  8. import page_utils
  9. def _create_train(conn, subjects):
  10. with st.form("train_form"):
  11. st.write("创建训练")
  12. position = st.selectbox("训练部位", ['左手', '右手'])
  13. finger_model = st.selectbox("气动手手势", list(page_utils.fingermodel_trans.keys()))
  14. trial_num = st.number_input("训练次数", value=10, step=5)
  15. owner_name = st.selectbox("用户", subjects.name.to_list())
  16. virtual_feedback_rate = st.number_input("假反馈比例", value=0., step=0.2)
  17. model_path = page_utils.file_selector(os.path.join(f'./model/{owner_name}'))
  18. submitted = st.form_submit_button("开始训练")
  19. if submitted:
  20. start_time = datetime.strptime(datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "%Y-%m-%d %H:%M:%S")
  21. train_new = {"position": position,
  22. "finger_model": page_utils.fingermodel_trans[finger_model],
  23. "trial_num": int(trial_num),
  24. "start_time": start_time,
  25. "owner_name": owner_name,
  26. "virtual_feedback_rate": float(virtual_feedback_rate),
  27. "model_path": model_path.name if model_path is not None else None}
  28. train.create_train(conn, train_new)
  29. os.system("python train_1.py")
  30. return owner_name
  31. def render():
  32. st.set_page_config(
  33. page_title="手势训练", page_icon=":chart_with_upwards_trend:"
  34. )
  35. hide_footer()
  36. st.markdown("# 手势训练")
  37. st.sidebar.success("训练")
  38. conn = st.connection("sql_app", type="sql")
  39. train.create_table(conn)
  40. subjects = subject.get_subjects(conn)
  41. sub_name = _create_train(conn, subjects)
  42. if sub_name:
  43. trains = train.get_trains(conn, sub_name)
  44. st.write("# 训练列表")
  45. st.dataframe(trains)
  46. render()