junyongyou / triq

TRIQ implementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

dataset

TANGLONG2 opened this issue · comments

mos_scale = [1, 2, 3, 4, 5]
image_files = {}
with open(mos_file, 'r+') as f:#打开文件为只读模式,文件指针位于文件开头
lines = f.readlines()#按行读取文件
for line in lines:
content = line.split(',')#将文件按行分成一个个数组
image_file = content[0].replace('"', '').lower()#取出数组第一列也就是文件中第一列图片名字

        if using_single_mos:
            score = float(content[-1]) if mos_format == 'mos' else float(content[1]) / 25. + 1

Hello,We have a puzzle. If use "single_mos" ,you have changed the MOS in the live-challenge data to [1-5], but the MOS tag in the LIVE data set in the code should be content [-1] instead of content [1]. We think the code should be
score = float(content[-1]) if mos_format == 'mos' else float(content[-1]) / 25. + 1

Hi, that depends on how you arrange the MOS scores in the file. The purpose is to just get a single MOS value from the file. So you can change to whatever format.