You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: dataset_collection_framework/README.md
+13-13Lines changed: 13 additions & 13 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -11,14 +11,14 @@ The two transmission flows (Wifi & bluetoot) are totally indipendent, a python s
11
11
The repository of the dataset framework collector is so structured:
12
12
1.**STM32 code** for the drone's state acquisition and bluetooth transmission -> `crazyflie_firmware_patch/`
13
13
2.**AI-deck code** for camera acquisition and wifi transmission -> `GAP8_streamer/`
14
-
2.**Python app** with the dataset collector framwork Graphical User Interface (GUI), which receives the images and state on the laptop, and label's post processing scripts -> `dataset_collector/`
14
+
2.**Python app** with the dataset collector framwork Graphical User Interface (GUI), which receives the images and state on the laptop, and label's post processing scripts -> `dataset_collector_python/`
To get the datasetcollector app running there are three main steps required:
@@ -27,9 +27,9 @@ To get the datasetcollector app running there are three main steps required:
27
27
28
28
**2. Flash the firmware on the AI Deck** -> `GAP8_streamer/`
29
29
30
-
**3. Open the dataset collector python app** -> `dataset_collector/`
30
+
**3. Open the dataset collector python app** -> `dataset_collector_python/`
31
31
32
-
**4. Collect a new dataset** -> stored in `dataset_collector/dataset/`
32
+
**4. Collect a new dataset** -> stored in `dataset_collector_python/dataset/`
33
33
34
34
### Prerequisites:
35
35
@@ -127,7 +127,7 @@ Now the AI Deck software is up and running, and it is streaming images via Wi-Fi
127
127
- To get the image streamer working some files inside the gap sdk hade been changed. The modified files are stored in `GAP8_streamer/gap_sdk_modified`. Therefore, the `Makefile` masks some files of the original gap_sdk to include these files that have been modified. By doing so, you don't have to modify the SDK.
128
128
129
129
**NINA firmware:**
130
-
we developed our application using some specific NINA firmware. If you buy a new AI deck is probably not compatible.
130
+
we developed our application using some specific NINA firmware. If you buy a new AI deck is probably not compatible.
131
131
132
132
Follow the [instructions here](https://github.com/LorenzoLamberti94/NINA-flashing-crazyflie).
Now we can open the Dataset Collector User interface with the following command: (**NOTE:** First you must plug in a joystick to control the crazyflie)
151
151
152
152
~~~~
153
-
cd dataset_collector/
153
+
cd dataset_collector_python/
154
154
conda activate dataset_collection_framework
155
155
python datasetcollector_main.py
156
156
~~~~
@@ -219,11 +219,11 @@ Left side of the Dataset collector GUI.
219
219
220
220
## Output of the dataset collector
221
221
222
-
-**The images will be saved in this folder:**`dataset_collector_framework/dataset_collector/dataset/acquisition#/images`
222
+
-**The images will be saved in this folder:**`dataset_collection_framework/dataset_collector_python/dataset/acquisition#/images`
223
223
224
-
-**The drone's state will be saved in this `csv` file:**`dataset_collector_framework/dataset_collector/dataset/acquisition#/state_labels_DroneState.csv`
224
+
-**The drone's state will be saved in this `csv` file:**`dataset_collection_framework/dataset_collector_python/dataset/acquisition#/state_labels_DroneState.csv`
225
225
226
-
**NOTE:** each acquisition will not overwrite the previous one. Each acquisition will create a new folder inside the `dataset_collector_framework/dataset_collector/dataset/` folder with consecutive numbering (`acquisition1`, `acquisition2`, ...).
226
+
**NOTE:** each acquisition will not overwrite the previous one. Each acquisition will create a new folder inside the `dataset_collection_framework/dataset_collector_python/dataset/` folder with consecutive numbering (`acquisition1`, `acquisition2`, ...).
227
227
228
228
# Post-processing the dataset labels
229
229
@@ -235,15 +235,15 @@ This script does so.
235
235
236
236
**Inputs:**
237
237
238
-
`--data_path` specifies the acquisition# in folder `dataset_collector_framework/dataset_collector/dataset/`. From this folder it authomatically parses:
238
+
`--data_path` specifies the acquisition# in folder `dataset_collection_framework/dataset_collector_python/dataset/`. From this folder it authomatically parses:
239
239
- the labels in `state_labels_DroneState.csv`
240
240
- all the images in the `images/` folder.
241
241
242
242
**Output:**
243
243
- A new .csv file `labeled_images.csv` with correspondence between image-and-state. Extra states collected are removed (the logging rate for the state is higher than the framerate of the images)
244
244
245
245
~~~~
246
-
cd /dataset_collector_framework/dataset_collector/dataset_tools/
246
+
cd /dataset_collection_framework/dataset_collector_python/dataset_tools/
0 commit comments