Rename grit.py to grit_generate.py

0236234f · echo840 · GitHub · fd7fc9ba · 0236234f
Unverified Commit 0236234f authored Jan 28, 2024 by echo840 Committed by GitHub Jan 28, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

data_generation/grit_generate.py data_generation/grit_generate.py +1 -1

No files found.
--- a/data_generation/grit.py
+++ b/data_generation/grit.py
@@ -129,4 +129,4 @@ if __name__ == "__main__":
            height, width = batch['image'][i].shape[0], batch['image'][i].shape[1]
            objects = dense_pred_to_normcaption(predictions[i], width, height)
            json_save.append({"img_id":batch['img_id'][i], "objects":objects})
-    save_json(json_save, args.output_path)
\ No newline at end of file
+    save_json(json_save, args.output_path)