kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Fix process_video bugs

Browse Source
This commit is contained in:
James Betker 2020-05-29 12:47:22 -06:00
parent beac71ad18
commit 5e9da65d81
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/archs/feature_arch.py
View File

@ -62,7 +62,3 @@ class WideResnetFeatureExtractor(nn.Module):
x = self.model.layer2(x) x = self.model.layer2(x)
x = self.model.layer3(x) x = self.model.layer3(x)
return x return x
w = WideResnetFeatureExtractor()
w.forward(torch.randn(3,64,64))

4
codes/process_video.py
View File

@ -38,8 +38,10 @@ class FfmpegBackedVideoDataset(data.Dataset):
def get_time_for_it(self, it): def get_time_for_it(self, it):
secs = it / self.frame_rate + self.start_at secs = it / self.frame_rate + self.start_at
mins = int(secs / 60) mins = int(secs / 60)
hours = int(mins / 60)
secs = secs - (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): def __getitem__(self, index):
if self.vertical_splits > 0: if self.vertical_splits > 0: