|
@@ -57,13 +57,13 @@ class ParamProcess(object):
|
|
obj = {
|
|
obj = {
|
|
"score": result[0],
|
|
"score": result[0],
|
|
"benchmark": 0.06,
|
|
"benchmark": 0.06,
|
|
- "is_good_video": result[0] > 0.06
|
|
|
|
|
|
+ "is_good_video": 1 if result[0] > 0.06 else 0
|
|
}
|
|
}
|
|
else:
|
|
else:
|
|
obj = {
|
|
obj = {
|
|
"score": None,
|
|
"score": None,
|
|
"benchmark": 0.06,
|
|
"benchmark": 0.06,
|
|
- "is_good_video": False
|
|
|
|
|
|
+ "is_good_video": 0
|
|
}
|
|
}
|
|
return obj
|
|
return obj
|
|
case "v2":
|
|
case "v2":
|
|
@@ -73,13 +73,13 @@ class ParamProcess(object):
|
|
obj = {
|
|
obj = {
|
|
"score": result[0],
|
|
"score": result[0],
|
|
"benchmark": 0.3,
|
|
"benchmark": 0.3,
|
|
- "is_good_video": result[0] > 0.3
|
|
|
|
|
|
+ "is_good_video": 1 if result[0] > 0.3 else 0
|
|
}
|
|
}
|
|
else:
|
|
else:
|
|
obj = {
|
|
obj = {
|
|
"score": None,
|
|
"score": None,
|
|
"benchmark": 0.3,
|
|
"benchmark": 0.3,
|
|
- "is_good_video": False
|
|
|
|
|
|
+ "is_good_video": 0
|
|
}
|
|
}
|
|
return obj
|
|
return obj
|
|
|
|
|
|
@@ -121,5 +121,5 @@ class ParamProcess(object):
|
|
:return:
|
|
:return:
|
|
"""
|
|
"""
|
|
version, features = await self.process_label(params)
|
|
version, features = await self.process_label(params)
|
|
- print(version, features)
|
|
|
|
|
|
+ # print(version, features)
|
|
return await self.predict_score(version, features)
|
|
return await self.predict_score(version, features)
|