Update README.md
Browse files
README.md
CHANGED
|
@@ -22,7 +22,7 @@ from detectron2 import model_zoo
|
|
| 22 |
import cv2
|
| 23 |
|
| 24 |
# edit all of these
|
| 25 |
-
image = cv2.imread("../../Datasets/inaturalist2021/custom2/val/imgs/
|
| 26 |
cfg = get_cfg()
|
| 27 |
cfg.merge_from_file(model_zoo.get_config_file('COCO-Detection/retinanet_R_101_FPN_3x.yaml'))
|
| 28 |
cfg.MODEL.WEIGHTS = './out3/model_0002899.pth'
|
|
|
|
| 22 |
import cv2
|
| 23 |
|
| 24 |
# edit all of these
|
| 25 |
+
image = cv2.imread("../../Datasets/inaturalist2021/custom2/val/imgs/01.jpg")
|
| 26 |
cfg = get_cfg()
|
| 27 |
cfg.merge_from_file(model_zoo.get_config_file('COCO-Detection/retinanet_R_101_FPN_3x.yaml'))
|
| 28 |
cfg.MODEL.WEIGHTS = './out3/model_0002899.pth'
|