mykcs / memsam-0415 Goto Github PK

/Users/myk/anaconda3/envs/seg-test-0401/bin/python /Users/myk/PyPjcts/MemSAM-0415/train_video.py 
/Users/myk/anaconda3/envs/seg-test-0401/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: dlopen(/Users/myk/anaconda3/envs/seg-test-0401/lib/python3.8/site-packages/torchvision/image.so, 0x0006): Symbol not found: __ZN3c1017RegisterOperatorsD1Ev
  Referenced from: <8E58E83E-9235-3324-9B6B-260614F85F69> /Users/myk/anaconda3/envs/seg-test-0401/lib/python3.8/site-packages/torchvision/image.so
  Expected in:     <77E8847E-95AD-3EF9-9314-BA1E83C408ED> /Users/myk/anaconda3/envs/seg-test-0401/lib/python3.8/site-packages/torch/lib/libtorch_cpu.dylib
  warn(f"Failed to load image Python extension: {e}")
Namespace(base_lr=0.0001, batch_size=1, disable_point_prompt=False, enable_memory=False, encoder_input_size=256, frame_length=10, keep_log=False, low_image_size=256, modelname='MemSAM', n_gpu=1, point_numbers=1, reinforce=False, sam_ckpt='checkpoints/sam_vit_b_01ec64.pth', semi=False, task='CAMUS_Video_Full', vit_name='vit_b', warmup=False, warmup_period=250)
Traceback (most recent call last):
  File "/Users/myk/PyPjcts/MemSAM-0415/train_video.py", line 229, in <module>
    main()
  File "/Users/myk/PyPjcts/MemSAM-0415/train_video.py", line 110, in main
    train_dataset = EchoVideoDataset(opt.data_path, opt.train_split, tf_train, img_size=args.encoder_input_size,frame_length=args.frame_length, point_numbers=args.point_numbers, disable_point_prompt=args.disable_point_prompt)
  File "/Users/myk/PyPjcts/MemSAM-0415/utils/data_us.py", line 810, in __init__
    with open(self.class_dict_file, 'r') as load_f:
FileNotFoundError: [Errno 2] No such file or directory: './dataset/SAMUS/CAMUS_full/class.json'

进程已结束，退出代码为 1