Commit 2419ad3b authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

make minor fixes

parent 4d117df3
......@@ -208,7 +208,7 @@ class Ava(object):
reader = csv.DictReader(annotations, fieldnames)
frame_annotations = collections.defaultdict(list)
ids = set()
# aggreggate by video and timestamp:
# aggregate by video and timestamp:
for row in reader:
ids.add(row["id"])
key = (row["id"], int(float(row["timestamp_seconds"])))
......@@ -284,15 +284,7 @@ class Ava(object):
logging.warning("Unknown label: %s", row["action_label"])
#Display the image and bounding boxes being
#processed (for debugging purposes)
#for i in range(len(xmins)):
# cv2.rectangle(image, (int(xmins[i] * width),
# int(ymaxs[i] * height)),
# (int(xmaxs[i] * width),
# int(ymins[i] * height)), (255, 0, 0), 2)
#cv2.imshow("mywindow", image)
#cv2.waitKey(1000)
#processed (for debugging purposes) if desired.
total_xmins.append(dataset_util.float_list_feature(xmins))
total_xmaxs.append(dataset_util.float_list_feature(xmaxs))
......@@ -332,7 +324,9 @@ class Ava(object):
'region/label/string':
feature_list_feature(total_label_strings),
'region/label/confidence':
feature_list_feature(total_confidences)
feature_list_feature(total_confidences), #all ones
'region/is_annotated':
feature_list_feature(total_confidences) #all ones
}
if len(total_xmins) > 0:
......
......@@ -16,7 +16,9 @@ done
rm "ava_file_names_trainval_v2.1.txt"
cd ..
#Trimming causes issues with frame seeking in the python script, so it is best left out.
# Trimming causes issues with frame seeking in the python script, so it is best left out.
# If included, need to modify the python script to subtract 900 seconds wheen seeking.
#echo "Trimming all videos."
#mkdir ava_vids_trimmed
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment