Skip to content

GitLab

Unverified Commit 0236234f authored by echo840's avatar echo840 Committed by GitHub
Browse files

Rename grit.py to grit_generate.py

parent fd7fc9ba
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
data_generation/grit.py → data_generation/grit_generate.py
View file @ 0236234f
...@@ -129,4 +129,4 @@ if __name__ == "__main__": ...@@ -129,4 +129,4 @@ if __name__ == "__main__":
height, width = batch['image'][i].shape[0], batch['image'][i].shape[1] height, width = batch['image'][i].shape[0], batch['image'][i].shape[1]
objects = dense_pred_to_normcaption(predictions[i], width, height) objects = dense_pred_to_normcaption(predictions[i], width, height)
json_save.append({"img_id":batch['img_id'][i], "objects":objects}) json_save.append({"img_id":batch['img_id'][i], "objects":objects})
save_json(json_save, args.output_path) save_json(json_save, args.output_path)
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment