Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion baselines/ec_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
warnings.simplefilter("ignore")
warnings.filterwarnings("ignore")

from qmap.qmap_model import QmapModel
from baselines.qmap.qmap_model import QmapModel
import ctypes
import copy
import json
Expand Down
4 changes: 2 additions & 2 deletions baselines/qmap/qmap_model.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import torch
import numpy as np
import qmap.models.models as models
import qmap.models.models_4x as models4x
import baselines.qmap.models.models as models
import baselines.qmap.models.models_4x as models4x
import time

from .utils import load_checkpoint
Expand Down
4 changes: 2 additions & 2 deletions h26x.py
Original file line number Diff line number Diff line change
Expand Up @@ -326,5 +326,5 @@ def run_one_file(index_file, output_dir, use_codec = "265"):
if mp.get_start_method(allow_none=True) is None:
mp.set_start_method('spawn', force=True)

run_one_file("INDEX_FINAL.txt", "results/h265", "265")
run_one_file("INDEX_FINAL.txt", "results/h264", "264")
run_one_file("INDEX.txt", "results/h265", "265")
run_one_file("INDEX.txt", "results/h264", "264")
2 changes: 1 addition & 1 deletion pretrained-gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -823,4 +823,4 @@ def run_one_file(index_file, output_dir):
final_df.to_csv(f"{output_dir}/all.csv", index=None)
return final_df

run_one_file("INDEX_FINAL.txt", "results/pretrained")
run_one_file("INDEX.txt", "results/pretrained")