From 5e9da65d813cde70d201e4967568649769989723 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 29 May 2020 12:47:22 -0600 Subject: [PATCH] Fix process_video bugs --- codes/models/archs/feature_arch.py | 6 +----- codes/process_video.py | 4 +++- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/codes/models/archs/feature_arch.py b/codes/models/archs/feature_arch.py index 3cafcc0d..80f77edb 100644 --- a/codes/models/archs/feature_arch.py +++ b/codes/models/archs/feature_arch.py @@ -61,8 +61,4 @@ class WideResnetFeatureExtractor(nn.Module): x = self.model.layer1(x) x = self.model.layer2(x) x = self.model.layer3(x) - return x - - -w = WideResnetFeatureExtractor() -w.forward(torch.randn(3,64,64)) \ No newline at end of file + return x \ No newline at end of file diff --git a/codes/process_video.py b/codes/process_video.py index df6046e8..c580543f 100644 --- a/codes/process_video.py +++ b/codes/process_video.py @@ -38,8 +38,10 @@ class FfmpegBackedVideoDataset(data.Dataset): def get_time_for_it(self, it): secs = it / self.frame_rate + self.start_at mins = int(secs / 60) + hours = int(mins / 60) secs = secs - (mins * 60) - return '%02d:%06.3f' % (mins, secs) + mins = mins % 60 + return '%02d:%02d:%06.3f' % (hours, mins, secs) def __getitem__(self, index): if self.vertical_splits > 0: