Spaces:
Sleeping
Sleeping
efengx
commited on
Commit
•
e1e83fc
1
Parent(s):
de5ad1a
fix: update
Browse files
fengxai/SelfSupervised.py
CHANGED
@@ -70,22 +70,20 @@ class SelfSupervised:
|
|
70 |
|
71 |
imageOutPutFile = "data/annotated_image.jpg"
|
72 |
fileList = imageOutPutFile.split("/")[0]
|
73 |
-
if not fileList:
|
|
|
74 |
os.mkdir(fileList)
|
75 |
cv2.imwrite(imageOutPutFile, imgAnnnotated)
|
76 |
print("os cwd=", os.getcwd())
|
77 |
for root, dirs, files in os.walk(os.getcwd()):
|
78 |
print("root=", root)
|
79 |
-
print("dirs=", dirs)
|
80 |
print("files=", files)
|
81 |
|
82 |
print("data=")
|
83 |
for root, dirs, files in os.walk("data/"):
|
84 |
print("root=", root)
|
85 |
-
print("dirs=", dirs)
|
86 |
print("files=", files)
|
87 |
-
|
88 |
-
|
89 |
return {
|
90 |
"imageOutput": imageOutPutFile,
|
91 |
"imageHeight": h,
|
|
|
70 |
|
71 |
imageOutPutFile = "data/annotated_image.jpg"
|
72 |
fileList = imageOutPutFile.split("/")[0]
|
73 |
+
if not os.path.exists(fileList):
|
74 |
+
print("fileList=", fileList)
|
75 |
os.mkdir(fileList)
|
76 |
cv2.imwrite(imageOutPutFile, imgAnnnotated)
|
77 |
print("os cwd=", os.getcwd())
|
78 |
for root, dirs, files in os.walk(os.getcwd()):
|
79 |
print("root=", root)
|
|
|
80 |
print("files=", files)
|
81 |
|
82 |
print("data=")
|
83 |
for root, dirs, files in os.walk("data/"):
|
84 |
print("root=", root)
|
|
|
85 |
print("files=", files)
|
86 |
+
|
|
|
87 |
return {
|
88 |
"imageOutput": imageOutPutFile,
|
89 |
"imageHeight": h,
|