This project models the nonlinear dynamics of a 2-DOF helicopter system using a Feedforward Neural Network (FNN). The model predicts the system's behavior based on input variables (voltage and motor currents) to output pitch and yaw angles.
You can access the Google Colab notebook for this project:
The model is trained and validated using real data representing the helicopter's behavior:
- Input Variables:
V: Voltage applied to the motors (constant for both motors).I_pitch: Current for the pitch motor.I_yaw: Current for the yaw motor.
- Output Variables:
pitch_angle (θ): The pitch angle of the helicopter.yaw_angle (ψ): The yaw angle of the helicopter.
The input and output data are stored in separate CSV files:
inputs.csv: Contains columnsV,I_pitch,I_yaw.outputs.csv: Contains columnspitch_angle,yaw_angle.
-
Ensure the trained model file
2dof_helicopter_model.h5is uploaded in colab. -
Use the model and change the inputs dat to make predictions:
new_input = np.array([[V, I_pitch, I_yaw]]) # Voltage, I_pitch, I_yaw