Hugging Face's logo

Spaces:
shikunl
/
prismer
Sleeping

App Files Community
7
shikunl commited on
Commit
6b33029
•
1 Parent(s): f1bf815

Update with md5sum and half precision inference

Browse files
Files changed (7) hide show
  1. prismer/configs/experts.yaml +1 -1
  2. prismer/experts/depth/generate_dataset.py +1 -1
  3. prismer/experts/edge/generate_dataset.py +1 -1
  4. prismer/experts/normal/generate_dataset.py +1 -1
  5. prismer/experts/obj_detection/generate_dataset.py +1 -1
  6. prismer/experts/ocr_detection/generate_dataset.py +1 -1
  7. prismer/experts/segmentation/generate_dataset.py +1 -1
prismer/configs/experts.yaml CHANGED
@@ -1,3 +1,3 @@
1
  data_path: helpers
2
- im_name: 87dfaeb4978ce05aa7be5e5b4cc1273a
3
  save_path: helpers/labels
 
1
  data_path: helpers
2
+ im_name: 4265e527d8897c032baabea78e590c18
3
  save_path: helpers/labels
prismer/experts/depth/generate_dataset.py CHANGED
@@ -17,7 +17,7 @@ class Dataset(Dataset):
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
- self.data_list = [f'prismer/helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
 
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
+ self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
prismer/experts/edge/generate_dataset.py CHANGED
@@ -17,7 +17,7 @@ class Dataset(Dataset):
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
- self.data_list = [f'prismer/helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
 
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
+ self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
prismer/experts/normal/generate_dataset.py CHANGED
@@ -17,7 +17,7 @@ class CustomDataset(Dataset):
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
- self.data_list = [f'prismer/helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
 
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
+ self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
prismer/experts/obj_detection/generate_dataset.py CHANGED
@@ -20,7 +20,7 @@ class Dataset(Dataset):
20
  self.data_path = config['data_path']
21
  self.depth_path = os.path.join(config['save_path'], 'depth', self.data_path.split('/')[-1])
22
  self.transform = transform
23
- self.data_list = [f'prismer/helpers/images/{config["im_name"]}.jpg']
24
 
25
  def __len__(self):
26
  return len(self.data_list)
 
20
  self.data_path = config['data_path']
21
  self.depth_path = os.path.join(config['save_path'], 'depth', self.data_path.split('/')[-1])
22
  self.transform = transform
23
+ self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
24
 
25
  def __len__(self):
26
  return len(self.data_list)
prismer/experts/ocr_detection/generate_dataset.py CHANGED
@@ -17,7 +17,7 @@ class Dataset(Dataset):
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
- self.data_list = [f'prismer/helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
 
17
  def __init__(self, config, transform):
18
  self.data_path = config['data_path']
19
  self.transform = transform
20
+ self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
21
 
22
  def __len__(self):
23
  return len(self.data_list)
prismer/experts/segmentation/generate_dataset.py CHANGED
@@ -19,7 +19,7 @@ class Dataset(Dataset):
19
  def __init__(self, config, transform):
20
  self.data_path = config['data_path']
21
  self.transform = transform
22
- self.data_list = [f'prismer/helpers/images/{config["im_name"]}.jpg']
23
 
24
  def __len__(self):
25
  return len(self.data_list)
 
19
  def __init__(self, config, transform):
20
  self.data_path = config['data_path']
21
  self.transform = transform
22
+ self.data_list = [f'helpers/images/{config["im_name"]}.jpg']
23
 
24
  def __len__(self):
25
  return len(self.data_list)