Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "9e4b2ff58eee5172c3ebaf8ca8cc1f1bcd49978d"
Commit cd174844 authored by Rahul Somani's avatar Rahul Somani Committed by Francisco Massa
Browse files

Removed code redundancy/refactored (#1549)

parent bb261c5c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
references/video_classification/train.py
View file @ cd174844
...@@ -144,13 +144,13 @@ def main(args): ...@@ -144,13 +144,13 @@ def main(args):
traindir, traindir,
frames_per_clip=args.clip_len, frames_per_clip=args.clip_len,
step_between_clips=1, step_between_clips=1,
transform=transform_train transform=transform_train,
frame_rate=15
) )
if args.cache_dataset: if args.cache_dataset:
print("Saving dataset_train to {}".format(cache_path)) print("Saving dataset_train to {}".format(cache_path))
utils.mkdir(os.path.dirname(cache_path)) utils.mkdir(os.path.dirname(cache_path))
utils.save_on_master((dataset, traindir), cache_path) utils.save_on_master((dataset, traindir), cache_path)
dataset.video_clips.compute_clips(args.clip_len, 1, frame_rate=15)
print("Took", time.time() - st) print("Took", time.time() - st)
...@@ -176,13 +176,13 @@ def main(args): ...@@ -176,13 +176,13 @@ def main(args):
valdir, valdir,
frames_per_clip=args.clip_len, frames_per_clip=args.clip_len,
step_between_clips=1, step_between_clips=1,
transform=transform_test transform=transform_test,
frame_rate=15
) )
if args.cache_dataset: if args.cache_dataset:
print("Saving dataset_test to {}".format(cache_path)) print("Saving dataset_test to {}".format(cache_path))
utils.mkdir(os.path.dirname(cache_path)) utils.mkdir(os.path.dirname(cache_path))
utils.save_on_master((dataset_test, valdir), cache_path) utils.save_on_master((dataset_test, valdir), cache_path)
dataset_test.video_clips.compute_clips(args.clip_len, 1, frame_rate=15)
print("Creating data loaders") print("Creating data loaders")
train_sampler = RandomClipSampler(dataset.video_clips, args.clips_per_video) train_sampler = RandomClipSampler(dataset.video_clips, args.clips_per_video)
......
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