-
Notifications
You must be signed in to change notification settings - Fork 125
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
StopIteration Error while training patches #6
Comments
In addition, ***/dm_keras_ext.py", line 304, |
Well, you shall use sample_patches_combined.py. |
Sorry, I have just made a mistake. The error was occured when I run patch_clf_train, not sample_patches_main. I have sampled both the mass and calc patches which divided into five categories: 'background', 'calc_mal', 'calc_ben', 'mass_mal', 'mass_ben'. But in the trainning model, It's divided into three classes: background, maligant, benign. Thus, I just combined the calc_mal and mass_mal to be malignant. It's the same as benign and background. |
I am also confused about the number of categories in patch_clf_train.py, It seems that the number should be 5 according to the paper. |
@ChaoLi977 |
hy .... did u able to run the complete code file? |
Hi,
I have sampled all the images to get the patches, and put them into the train, val, and test directory, each directory has three subdirectories: backround, benign and malignant patches. But when i run the sample_patches_main.py, i get the error. Can you help me solve the problem? Thanks a lot!
Here is the error:
Traceback (most recent call last):
File "patch_clf_train.py", line 313, in
run(args.train_dir, args.val_dir, args.test_dir, run_opts)
File "patch_clf_train.py", line 155, in run
hidden_dropout2=hidden_dropout2)
File "/home//end2end-all-conv-master/ddsm_train/dm_keras_ext.py", line 304, in do_3stage_training
verbose=2)
File "/home//anaconda2/envs/tensorflow/lib/python2.7/site-packages/keras/legacy/interfaces.py", line 91, in wrapper
return func(*args, kwargs)
File "/home//anaconda2/envs/tensorflow/lib/python2.7/site-packages/keras/engine/training.py", line 2192, in fit_generator
generator_output = next(output_generator)
File "/home//anaconda2/envs/tensorflow/lib/python2.7/site-packages/keras/utils/data_utils.py", line 584, in get
six.raise_from(StopIteration(e), e)
File "/home/**/anaconda2/envs/tensorflow/lib/python2.7/site-packages/six.py", line 737, in raise_from
raise value
StopIteration
The text was updated successfully, but these errors were encountered: