fix the use of cfg in workers

tensorpack · Oct 19, 2019 · 74badc6 · 74badc6
1 parent a6ca79c
commit 74badc6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/FasterRCNN/data.py b/examples/FasterRCNN/data.py
@@ -99,7 +99,8 @@ def __call__(self, roidb):
         points = tfms.apply_coords(points)
         boxes = point8_to_box(points)
         if len(boxes):
-            assert klass.max() <= cfg.DATA.NUM_CATEGORY, "Invalid category {}!".format(klass.max())
+            assert klass.max() <= self.cfg.DATA.NUM_CATEGORY, \
+                "Invalid category {}!".format(klass.max())
             assert np.min(np_area(boxes)) > 0, "Some boxes have zero area!"
 
         ret = {"image": im}