이현규

Add final paper, Add several featuremaps, etc

......@@ -8,8 +8,7 @@ import esot3ria.video_recommender as recommender
import esot3ria.video_util as videoutil
# Define file paths.
MODEL_PATH = "/Users/esot3ria/PycharmProjects/yt8m/models/frame/" \
"refined_model/inference_model/segment_inference_model"
MODEL_PATH = "/Users/esot3ria/PycharmProjects/yt8m/models/frame/refined_model/inference_model/segment_inference_model"
VOCAB_PATH = "../vocabulary.csv"
VIDEO_TAGS_PATH = "./kaggle_solution_40k.csv"
TAG_VECTOR_MODEL_PATH = "./tag_vectors.model"
......@@ -218,6 +217,6 @@ def inference_pb(file_path):
if __name__ == '__main__':
filepath = "features.pb"
filepath = "../featuremaps/features(yorusika).pb"
result = inference_pb(filepath)
print(result)
......
......@@ -335,7 +335,11 @@ def inference(reader, train_dir, data_pattern, out_file_location, batch_size,
cls_score_dict = {}
out_file.seek(0, 0)
old_seg_name = '0000'
counter = 0
for line in out_file:
counter += 1
if counter / 5000 == 0:
print(counter, " processed")
segment_id, preds = line.decode("utf8").split(",")
if segment_id == "VideoId":
# Skip the headline.
......