diff --git a/README.md b/README.md index 1764412..a6571e9 100644 --- a/README.md +++ b/README.md @@ -26,7 +26,7 @@ conda env create -f ./environment/haheae.yaml -n haheae conda activate haheae ``` -Step 2: Follow the instructions at the [Pose2Gaze project][1] to process the datasets. +Step 2: Follow the instructions at [Pose2Gaze][1] to process the datasets. Step 3: Set 'data_dir' in 'config.py' and 'main.py' for the processed datasets. Run 'train.sh' to evaluate the pre-trained models. If you want to train the model from scratch, you can remove the pre-trained models and uncomment the training command (the command with "mode" set to "train").