haritsahm
commited on
Commit
·
5d412e5
1
Parent(s):
7011045
Hotfix missing examples on docker and filter files
Browse files- .dockerignore +1 -0
- main.py +4 -1
.dockerignore
CHANGED
@@ -2,5 +2,6 @@
|
|
2 |
!models/
|
3 |
!weights/
|
4 |
!utils/
|
|
|
5 |
!main.py
|
6 |
!requirements.txt
|
|
|
2 |
!models/
|
3 |
!weights/
|
4 |
!utils/
|
5 |
+
!examples/
|
6 |
!main.py
|
7 |
!requirements.txt
|
main.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import os
|
|
|
2 |
from pathlib import Path
|
3 |
from typing import List
|
4 |
|
@@ -27,7 +28,9 @@ SUPPORTED_IMG_EXT = ['.png', '.jpg', '.jpeg']
|
|
27 |
|
28 |
EXAMPLE_IMAGES = [
|
29 |
os.path.join(os.path.dirname(__file__), f'examples/{f}')
|
30 |
-
for f in os.listdir('examples/')
|
|
|
|
|
31 |
|
32 |
|
33 |
def filter_files(files: List) -> List:
|
|
|
1 |
import os
|
2 |
+
import re
|
3 |
from pathlib import Path
|
4 |
from typing import List
|
5 |
|
|
|
28 |
|
29 |
EXAMPLE_IMAGES = [
|
30 |
os.path.join(os.path.dirname(__file__), f'examples/{f}')
|
31 |
+
for f in os.listdir('examples/')
|
32 |
+
if re.search(r'.*.(jpg|jpeg|tiff|JPEG|PNG|png)', f)
|
33 |
+
]
|
34 |
|
35 |
|
36 |
def filter_files(files: List) -> List:
|