Skip to content
Snippets Groups Projects
Commit 1e9f3e89 authored by Levecque Etienne's avatar Levecque Etienne
Browse files

fix: correct ilp search with color image

parent 1b616491
No related branches found
No related tags found
No related merge requests found
......@@ -163,7 +163,7 @@ class Image:
def search_antecedent_ilp(self, parameters, shared_dict, task_id, verbose):
if (not self.is_jpeg
or not isinstance(self.pipeline, NaivePipeline)
or not np.allclose(self.pipeline.quant_tbl, np.ones((8, 8)))):
or not np.allclose(self.pipeline.quant_tbl, np.ones(self.pipeline.quant_tbl.shape))):
raise ValueError('Gurobi search is only effective for JPEG images at QF100 with the naive pipeline.')
self.filter_blocks()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment