Spaces:
Build error
Build error
Commit
·
7845cb9
1
Parent(s):
3362450
Update app.py
Browse files
app.py
CHANGED
@@ -8,9 +8,6 @@ import cv2
|
|
8 |
from detectron2.config import get_cfg
|
9 |
from utils import add_bboxes
|
10 |
|
11 |
-
# print(torch.__version__, torch.cuda.is_available())
|
12 |
-
# assert torch.__version__.startswith("1.9")
|
13 |
-
|
14 |
config_file="config.yaml"
|
15 |
cfg = get_cfg()
|
16 |
cfg.merge_from_file(config_file)
|
@@ -37,7 +34,7 @@ def predict(
|
|
37 |
return img
|
38 |
|
39 |
title = "Indoor Pet Detection"
|
40 |
-
description = "This is an application trained with synthetic data from Unity Computer Vision. We trained a single class object detection model to recognize dogs using images of pets randomly posed and placed in our home interior environment. There are multiple pre-trained models trained with 5k, 40k and 100k synthetic data that you can choose to see the results."
|
41 |
examples = [
|
42 |
["5k synthetic", 'example.jpg'],
|
43 |
["40k synthetic", 'example.jpg'],
|
|
|
8 |
from detectron2.config import get_cfg
|
9 |
from utils import add_bboxes
|
10 |
|
|
|
|
|
|
|
11 |
config_file="config.yaml"
|
12 |
cfg = get_cfg()
|
13 |
cfg.merge_from_file(config_file)
|
|
|
34 |
return img
|
35 |
|
36 |
title = "Indoor Pet Detection"
|
37 |
+
description = "This is an application trained with synthetic data from Unity Computer Vision. We trained a single class object detection model to recognize dogs using images of pets randomly posed and placed in our home interior environment. There are multiple pre-trained models trained with 5k, 40k and 100k synthetic data that you can choose to see the results. <p> <a href='https://github.com/Unity-Technologies/Indoor-Pet-Detection'> Github Project </a> </p>"
|
38 |
examples = [
|
39 |
["5k synthetic", 'example.jpg'],
|
40 |
["40k synthetic", 'example.jpg'],
|