Skip to content

Instantly share code, notes, and snippets.

@zaidbhat1234
Last active July 12, 2021 19:08
Show Gist options
  • Save zaidbhat1234/ed92df30f8bd5c768be1c32d257f0c4b to your computer and use it in GitHub Desktop.
Save zaidbhat1234/ed92df30f8bd5c768be1c32d257f0c4b to your computer and use it in GitHub Desktop.
Load Dataset
from autovideo.utils import set_log_path
import pandas as pd
import os
set_log_path('log.txt') #Setup logger
data_dir = 'datasets/hmdb6/' #Directory containing dataset (hmdb-6)
train_table_path = os.path.join(data_dir, 'train.csv')
train_media_dir = os.path.join(data_dir, 'media')
target_index = 2 #Index of column containing label information
# Read the CSV file
train_dataset = pd.read_csv(train_table_path)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment