shikunl commited on
Commit
f1bf815
β€’
1 Parent(s): 0163f06

Update with md5sum and half precision inference

Browse files
Dockerfile CHANGED
@@ -38,7 +38,7 @@ RUN pyenv install ${PYTHON_VERSION} && \
38
  pyenv rehash && \
39
  pip install --no-cache-dir -U pip setuptools wheel
40
 
41
- RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
42
  COPY --chown=1000 requirements.txt /tmp/requirements.txt
43
  RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
44
 
 
38
  pyenv rehash && \
39
  pip install --no-cache-dir -U pip setuptools wheel
40
 
41
+ RUN pip install --no-cache-dir -U torch==2.0.0 torchvision==0.15.1
42
  COPY --chown=1000 requirements.txt /tmp/requirements.txt
43
  RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
44
 
prismer/download_checkpoints.py CHANGED
@@ -30,7 +30,7 @@ _MODELS = [
30
  "pretrain_prismerz_large",
31
  ]
32
 
33
- _REPO_ID = "lorenmt/prismer"
34
 
35
 
36
  def download_checkpoints(
 
30
  "pretrain_prismerz_large",
31
  ]
32
 
33
+ _REPO_ID = "shikunl/prismer"
34
 
35
 
36
  def download_checkpoints(
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'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'prismer/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'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'prismer/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'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'prismer/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'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'prismer/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'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'prismer/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'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'prismer/helpers/images/{config["im_name"]}.jpg']
23
 
24
  def __len__(self):
25
  return len(self.data_list)