search

babbu3682 / SMART-Net

Official SMART-Net Code
MIT License
18 stars 3 forks source link

Dataset error #1

Open danny0628 opened 1 year ago

danny0628 commented 1 year ago

Hello, @babbu3682 Thank you for providing code! I try to train custom dataset, but there is an error. (also testing your sample dataset, but occur same error) my train.sh code is 

python train.py \
--data-folder-dir '/mnt/home/mjkim1/node6.gpu/SMART-Net/datasets/samples' \
--model-name 'Up_SMART_Net' \
--batch-size 16 \
--epochs 1000 \
--num-workers 4 \
--pin-mem \
--training-stream 'Upstream' \
--multi-gpu-mode 'DataParallel' \
--cuda-visible-devices '2, 3' \
--gradual-unfreeze 'True' \
--print-freq 1 \
--output-dir '/mnt/home/mjkim1/node6.gpu/SMART-Net/ckpt'

and, this is my error

Original Traceback (most recent call last):                                                                                                                                                                                                                                                                                                                                         
  File "/home/mjkim1/miniconda3/envs/smartnet/lib/python3.8/site-packages/monai/transforms/transform.py", line 141, in apply_transform                                                    
    return _apply_transform(transform, data, unpack_items, lazy, overrides, log_stats)                                                                                                    
  File "/home/mjkim1/miniconda3/envs/smartnet/lib/python3.8/site-packages/monai/transforms/transform.py", line 98, in _apply_transform                                                                                                                                                                                                                                              
    return transform(data, lazy=lazy) if isinstance(transform, LazyTrait) else transform(data)                                                                                            
  File "/mnt/home/mjkim1/node6.gpu/SMART-Net/create_datasets/Hemorrhage.py", line 161, in Albu_2D_Transform_Compose                                                                       
    augment = Trans(image=image, mask=mask)                                                                                                                                                                                                                                                                                                                                         
  File "/home/mjkim1/miniconda3/envs/smartnet/lib/python3.8/site-packages/albumentations/core/composition.py", line 195, in __call__                                                      
    self._check_args(**data)                                                                                                                                                                                                                                                                                                                                                        
  File "/home/mjkim1/miniconda3/envs/smartnet/lib/python3.8/site-packages/albumentations/core/composition.py", line 275, in _check_args                                                   
    raise TypeError("{} must be numpy array type".format(data_name))                                                                                                                      
TypeError: image must be numpy array type

I use monai version 1.1.0

babbu3682 commented 6 days ago

Sorry for the late answer. :( We have updated all the learning and evaluation codes, please refer to them.