robot-surgery-segmentation
robot-surgery-segmentation copied to clipboard
Can't use the UNet11 and UNet 16 in generate_mask.py
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 ?
Have you passed the parameter --model_type
properly?
e.g.
python generate_masks.py --output_path predictions/AlbuNet/work --model_type UNet16