search

SUYEgit / Surgery-Robot-Detection-Segmentation

Object detection and segmentation for a surgery robot using Mask-RCNN on Python 3, Keras, and TensorFlow..
Other
181 stars 83 forks source link

Prediction.ipynb errors #14

Open amullapu opened 5 years ago

amullapu commented 5 years ago

I tried to run the inferenceconfig function and I get the following error

NameError Traceback (most recent call last)

in () ----> 1 class InferenceConfig(surgery.SurgeryConfig): 2 # Set batch size to 1 since we'll be running inference on 3 # one image at a time. Batch size = GPU_COUNT * IMAGES_PER_GPU 4 GPU_COUNT = 1 5 IMAGES_PER_GPU = 1 NameError: name 'surgery' is not defined