intisarhasnain commited on
Commit
28ff704
·
1 Parent(s): d558fc2

fix: correct weights filename to faster_rcnn_latest.pth

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), "mmdetection"))
9
  from mmdet.apis import init_detector, inference_detector
10
 
11
  CONFIG_FILE = "configs/faster_rcnn.py"
12
- WEIGHTS_FILE = "weights/faster_rcnn.pth"
13
  DEVICE = "cpu"
14
  CLASS_NAMES = ["wall", "room"]
15
  CLASS_COLORS = {
 
9
  from mmdet.apis import init_detector, inference_detector
10
 
11
  CONFIG_FILE = "configs/faster_rcnn.py"
12
+ WEIGHTS_FILE = "weights/faster_rcnn_latest.pth"
13
  DEVICE = "cpu"
14
  CLASS_NAMES = ["wall", "room"]
15
  CLASS_COLORS = {