Skip to content

Commit

Permalink
add script info
Browse files Browse the repository at this point in the history
hanzheteng committed Nov 2, 2023
1 parent d35b6c9 commit 8b8d826
Showing 3 changed files with 38 additions and 0 deletions.
18 changes: 18 additions & 0 deletions scripts/bag2files.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,21 @@
# File: bag2files.py
# Authors: Hanzhe Teng et al.
# Date: 2023-10-31
# Description: This script is part of the CitrusFarm Dataset (https://ucr-robotics.github.io/Citrus-Farm-Dataset/),
# which is released under the Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) license.

"""
This script extracts data from rosbags and saves them to individual files.
Usage: python bag2files.py <src_folder> <output_folder> (e.g. python bag2files.py 01_13B_Jackal /path/to/output)
<src_folder> is the folder containing rosbags to be processed
<output_folder> is the folder to save the extracted data
You can also modify the following two variables to select those rosbags and topics that you want to process:
rosbag_prefixes_of_interest
known_topic_info
"""

import argparse
import csv
import os
10 changes: 10 additions & 0 deletions scripts/download_citrusfarm.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# File: download_citrusfarm.py
# Authors: Hanzhe Teng et al.
# Date: 2023-10-11
# Description: This script is part of the CitrusFarm Dataset (https://ucr-robotics.github.io/Citrus-Farm-Dataset/),
# which is released under the Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) license.

"""
This script downloads the CitrusFarm Dataset from the AWS S3 bucket and verifies the MD5 checksums.
"""

import wget
import os
import yaml
10 changes: 10 additions & 0 deletions scripts/gen_dataset_file_list.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,13 @@
# File: gen_dataset_file_list.py
# Authors: Hanzhe Teng et al.
# Date: 2023-10-11
# Description: This script is part of the CitrusFarm Dataset (https://ucr-robotics.github.io/Citrus-Farm-Dataset/),
# which is released under the Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) license.

"""
This script generates the dataset_file_list.yaml file, which contains the MD5 checksums of all files in the dataset.
"""

import os
import hashlib
import yaml

0 comments on commit 8b8d826

Please sign in to comment.