efengx commited on
Commit
e1e83fc
1 Parent(s): de5ad1a

fix: update

Browse files
Files changed (1) hide show
  1. fengxai/SelfSupervised.py +3 -5
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,