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
@yjh0410 can you provide guildline how to train YOWOv2 on my own dataset here is alteration i make on dataset config file
'ava_v2.2': {
# dataset
'frames_dir': '/home/ubuntu/Furqan/new_dataset/20231004-activity-all-annotated-tasks/',
'frame_list': '/home/ubuntu/hammad/YOWOv2/config/frame/',
'annotation_dir': '/home/ubuntu/hammad/new_annotations/',
'train_gt_box_list': '/home/ubuntu/hammad/new_annotations/20231005_ke_activity_train_avastyle_17.csv',
'val_gt_box_list': '/home/ubuntu/hammad/new_annotations/20231005_ke_activity_val_avastyle_17.csv',
'train_exclusion_file': '/home/ubuntu/hammad/new_annotations/ava_train_excluded_timestamps_v2.2 (1).csv',
'val_exclusion_file': '/home/ubuntu/hammad/new_annotations/ava_val_excluded_timestamps_v2.2.csv',
'labelmap_file': '/home/ubuntu/hammad/YOWOv2/labels.pbtxt', # 'ava_v2.2/ava_action_list_v2.2.pbtxt',
'class_ratio_file': None,
'backup_dir': 'results/',
**however it give me this error**
Model Config: YOWO_V2_LARGE
Finished loading image paths from: /home/ubuntu/hammad/YOWOv2/config/frame/train.csv
Traceback (most recent call last):
File "train.py", line 330, in
train()
File "train.py", line 156, in train
dataset, evaluator, num_classes = build_dataset(d_cfg, args, is_train=True)
File "/home/ubuntu/hammad/YOWOv2/utils/misc.py", line 67, in build_dataset
dataset = AVA_Dataset(
File "/home/ubuntu/hammad/YOWOv2/dataset/ava.py", line 49, in init
self._load_data()
File "/home/ubuntu/hammad/YOWOv2/dataset/ava.py", line 65, in _load_data
boxes_and_labels = ava_helper.load_boxes_and_labels(
File "/home/ubuntu/hammad/YOWOv2/dataset/ava_helper.py", line 134, in load_boxes_and_labels
if box_key not in all_boxes[video_name][frame_sec]:
KeyError: 0
The text was updated successfully, but these errors were encountered:
@yjh0410 can you provide guildline how to train YOWOv2 on my own dataset here is alteration i make on dataset config file
'ava_v2.2': {
# dataset
'frames_dir': '/home/ubuntu/Furqan/new_dataset/20231004-activity-all-annotated-tasks/',
'frame_list': '/home/ubuntu/hammad/YOWOv2/config/frame/',
'annotation_dir': '/home/ubuntu/hammad/new_annotations/',
'train_gt_box_list': '/home/ubuntu/hammad/new_annotations/20231005_ke_activity_train_avastyle_17.csv',
'val_gt_box_list': '/home/ubuntu/hammad/new_annotations/20231005_ke_activity_val_avastyle_17.csv',
'train_exclusion_file': '/home/ubuntu/hammad/new_annotations/ava_train_excluded_timestamps_v2.2 (1).csv',
'val_exclusion_file': '/home/ubuntu/hammad/new_annotations/ava_val_excluded_timestamps_v2.2.csv',
'labelmap_file': '/home/ubuntu/hammad/YOWOv2/labels.pbtxt', # 'ava_v2.2/ava_action_list_v2.2.pbtxt',
'class_ratio_file': None,
'backup_dir': 'results/',
Model Config: YOWO_V2_LARGE
Finished loading image paths from: /home/ubuntu/hammad/YOWOv2/config/frame/train.csv
Traceback (most recent call last):
File "train.py", line 330, in
train()
File "train.py", line 156, in train
dataset, evaluator, num_classes = build_dataset(d_cfg, args, is_train=True)
File "/home/ubuntu/hammad/YOWOv2/utils/misc.py", line 67, in build_dataset
dataset = AVA_Dataset(
File "/home/ubuntu/hammad/YOWOv2/dataset/ava.py", line 49, in init
self._load_data()
File "/home/ubuntu/hammad/YOWOv2/dataset/ava.py", line 65, in _load_data
boxes_and_labels = ava_helper.load_boxes_and_labels(
File "/home/ubuntu/hammad/YOWOv2/dataset/ava_helper.py", line 134, in load_boxes_and_labels
if box_key not in all_boxes[video_name][frame_sec]:
KeyError: 0
The text was updated successfully, but these errors were encountered: