register annotation file with points internally

Alexander Kirillov · facebook-github-bot · commit 40f8cffa7eb9 · 2021-07-22T13:11:19.000-07:00
Summary: make point supervision compatible with internal dataset file system.

Differential Revision: D29629927

fbshipit-source-id: 5646645d118ac82816269b7a16d2328efccaf183
diff --git a/projects/PointSup/point_sup/register_point_annotations.py b/projects/PointSup/point_sup/register_point_annotations.py
@@ -62,5 +62,8 @@ def register_all_coco_train_points(root):
             )
 
 
-_root = os.getenv("DETECTRON2_DATASETS", "datasets")
-register_all_coco_train_points(_root)
+# True for open source;
+# Internally at fb, we register them elsewhere
+if __name__.endswith(".register_point_annotations"):
+    _root = os.getenv("DETECTRON2_DATASETS", "datasets")
+    register_all_coco_train_points(_root)

Original file line number	Diff line number	Diff line change
`@@ -62,5 +62,8 @@ def register_all_coco_train_points(root):`
`62`	`62`	`)`
`63`	`63`
`64`	`64`
`65`		`-_root = os.getenv("DETECTRON2_DATASETS", "datasets")`
`66`		`-register_all_coco_train_points(_root)`
	`65`	`+# True for open source;`
	`66`	`+# Internally at fb, we register them elsewhere`
	`67`	`+if __name__.endswith(".register_point_annotations"):`
	`68`	`+ _root = os.getenv("DETECTRON2_DATASETS", "datasets")`
	`69`	`+ register_all_coco_train_points(_root)`