Skip to content
Snippets Groups Projects
Commit aa239aa4 authored by Franck Galpin's avatar Franck Galpin
Browse files

Draft: Fix mr94 data conversion

parent 89a157b4
No related branches found
No related tags found
No related merge requests found
......@@ -110,10 +110,10 @@ def readDataInt16(
data_h, w
)
data = data[:, srcc[2] : srcc[3]]
t[dstc[0] : dstc[1], dstc[2] : dstc[3]] = data * norm
t[dstc[0] : dstc[1], dstc[2] : dstc[3]] = data
if scale:
t = t.repeat(2, axis=0).repeat(2, axis=1)
return t.astype("int16")
return t.astype("int16") * norm
def readOne(patch_size, border_size, norm, fn, off, ctype):
......@@ -169,9 +169,7 @@ def hasLuma(comp):
class DataLoader:
components = []
database = None # contains the whole database
patch_info = (
None
) # contains address of each patch in the database: dataset index, frame index in the dataset, patch_index in the frame
patch_info = None # contains address of each patch in the database: dataset index, frame index in the dataset, patch_index in the frame
suffix = {} # suffix for each file
# patch_size in luma sample
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment