Commit ·
28ff704
1
Parent(s): d558fc2
fix: correct weights filename to faster_rcnn_latest.pth
Browse files
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/
|
| 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 = {
|