ZiyuG commited on
Commit
5c6d6ba
1 Parent(s): 1b44373

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +9 -9
evaluate.py CHANGED
@@ -254,21 +254,21 @@ def eval(test, standard, tmpdir):
254
  return sum(scores) / len(scores), final_merged_intervals, comments
255
 
256
  def install():
257
- if torch.cuda.is_available():
258
- cu_version = torch.version.cuda
259
- cu_version = f"cu{cu_version.replace('.', '')}" # Format it as 'cuXX' (e.g., 'cu113')
260
- else:
261
- cu_version = "cpu" # Fallback to CPU if no CUDA is available
262
 
263
- torch_version = torch.__version__.split('+')[0] # Get PyTorch version without build info
264
 
265
- pip_command = f'pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html'
266
 
267
 
268
- os.system(pip_command)
269
 
270
  os.system('mim install mmengine')
271
- # # os.system('mim install "mmcv>=2.0.1"')
272
  os.system('mim install "mmdet>=3.1.0"')
273
  os.system('git clone https://github.com/open-mmlab/mmpose.git')
274
  os.chdir('mmpose')
 
254
  return sum(scores) / len(scores), final_merged_intervals, comments
255
 
256
  def install():
257
+ # if torch.cuda.is_available():
258
+ # cu_version = torch.version.cuda
259
+ # cu_version = f"cu{cu_version.replace('.', '')}" # Format it as 'cuXX' (e.g., 'cu113')
260
+ # else:
261
+ # cu_version = "cpu" # Fallback to CPU if no CUDA is available
262
 
263
+ # torch_version = torch.__version__.split('+')[0] # Get PyTorch version without build info
264
 
265
+ # pip_command = f'pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html'
266
 
267
 
268
+ # os.system(pip_command)
269
 
270
  os.system('mim install mmengine')
271
+ os.system('mim install "mmcv>=2.0.1"')
272
  os.system('mim install "mmdet>=3.1.0"')
273
  os.system('git clone https://github.com/open-mmlab/mmpose.git')
274
  os.chdir('mmpose')