2.9 KiB
Data Collection: Quick Demo
If you just want to run the completed project in order to collect your training and validation data this section can help do it.
To learn how to build something like this from scratch, see Part 1 and Part 2 of our tutorial.
Table of Contents
Prerequisites
To follow this tutorial you need to clone this repository even if you want to create your Unity project from scratch.
Note For Windows Users: You need to have a software enabling you to run bash files. One option is to download GIT. During installation of GIT, add GIT Bash to windows context menu by selecting its option. After installation right click in your folder select GIT Bash Here.
- Open a terminal and put yourself where you want to host the repository.
git clone https://github.com/Unity-Technologies/Unity-Robotics-Hub.git
-
Open the completed project. To do so, open Unity Hub, click the
Add
button, and selectPoseEstimationDemoProject
from theUnity-Robotics-Hub/tutorials/pose_estimation/
folder.
Setup
-
Open the scene. Go to
Assets > Scenes
and double click onTutorialPoseEstimation
. -
The size of the images that will be used for pose estimation depends on a setting in the Game view. Select the
Game
view and selectFree Aspect
. Then select the +, with the messageAdd new item
on it if you put your mouse over the + sign. For the Width select650
and for the Height select400
. A gif below shows you how to do it.
Switching to Data Collection Mode
The completed project is set up for inference mode by default, so we must switch it to data collection mode.
-
Uncheck the
ROSObjects
GameObject in the Hierarchy tab to disable it. -
On the
Simulation Scenario
GameObject, check theFixed Length Scenario
component to enable it. -
On the
Main Camera
GameObject, check thePerception Camera (Script)
component to enable it.
Data Collection
To get strarted with the data collection, follow the instructions in Part 3: Collect the Training and Validation Data of the tutorial. This section will explain how to set the random seed of the environment, choose how many training data examples you'd like to collect, and get it running.
If you'd like to then move on to training a pose estimation model on the data you've collected, move on to Part 3: Train the Deep Learning Model.
Have fun!