Skip to content
This repository was archived by the owner on Sep 23, 2024. It is now read-only.

Commit 9e3f881

Browse files
authored
Merge pull request #252 from FIRST-Tech-Challenge/pr_issue_251
Handle case where a video has exactly one included frame.
2 parents 4fa53f2 + 7c19005 commit 9e3f881

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

server/dataset_producer.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -175,13 +175,7 @@ def __split_for_records(video_frame_entities, eval_percent, max_frames_per_recor
175175
label_set.update(set(labels))
176176
random.shuffle(included_frame_numbers)
177177

178-
included_frame_count = len(included_frame_numbers)
179-
if included_frame_count == 1 and eval_percent > 0 and eval_percent < 100:
180-
message = "Error: if the number of included video frames is 1, eval_percent must be 0 or 100."
181-
logging.critical(message)
182-
raise exceptions.HttpErrorUnprocessableEntity(message)
183-
184-
if eval_percent == 0:
178+
if eval_percent == 0 or len(included_frame_numbers) == 1:
185179
eval_frame_numbers = []
186180
train_frame_numbers = included_frame_numbers
187181
elif eval_percent == 100:

0 commit comments

Comments
 (0)