robot-surgery-segmentation icon indicating copy to clipboard operation
robot-surgery-segmentation copied to clipboard

Can't use the UNet11 and UNet 16 in generate_mask.py

Open King-king424 opened this issue 6 years ago • 1 comments

when I run the generate_mask.py,I found that UNet11 and UNet16 can not be used. I don't know why.Anybody can help me ?

King-king424 avatar Jan 23 '19 16:01 King-king424

Have you passed the parameter --model_type properly?

e.g. python generate_masks.py --output_path predictions/AlbuNet/work --model_type UNet16

sakares avatar Mar 23 '19 03:03 sakares