The official repo of paper 'Automatic Truss Design with Reinforcement Learning'

The generated trusses are in folder 'assets', you can evaluate them by following command:

python apps/eval.py --config $TRUSS_CONFIG$ --draw-file $TRUSS_PATH$

For example:

python apps/eval.py --config 17_bar_case --draw-file assets/17_bar_1378.txt

This command will generate a truss image.

./run.sh without_buckle_case1 test

The command will generate truss automatically and save the truss in 'PostResults/without_buckle_case1/test/'

Or use the following instructions sequentially:

Generating truss layouts in stage 1.

python ./Stage1/continuous_uct_3d.py --config without_buckle_case1 --run-id test

Change output's format to input for stage 2.

python ./Stage1/noise_input_permutation_format_transfer.py --config without_buckle_case1 --run-id test

Use RL to search for lighter truss layouts.

python ./Stage2/main_3d.py --config without_buckle_case1 --run-id test

To generate your own truss with different configs, you can add your own config file in 'configs' and register it in 'config.py'.