(basedir)
| 50 | |
| 51 | |
| 52 | def register_balloon(basedir): |
| 53 | for split in ["train", "val"]: |
| 54 | name = "balloon_" + split |
| 55 | DatasetRegistry.register(name, lambda x=split: BalloonDemo(basedir, x)) |
| 56 | DatasetRegistry.register_metadata(name, "class_names", ["BG", "balloon"]) |
| 57 | |
| 58 | |
| 59 | if __name__ == '__main__': |
no test coverage detected