search

maju116 / pyplatypus

Python package for object detection and image segmentation.
MIT License
2 stars 0 forks source link

Prepare Platypus to a DEMO #41

Closed borkowski1110 closed 2 years ago

borkowski1110 commented 2 years ago
  1. Implement .predict() method.
  2. Implement plotting: original image plus original masks plus predicted one.
  3. Generated images to the folders: original_data_source/image/predicted_masks/model1_prediction.jpg (predict)
  4. Revert the split_to_binary to prepare the image for plotting. Add the choosing of the class based on the highest prob.
borkowski1110 commented 2 years ago

Images here: https://www.kaggle.com/competitions/siim-acr-pneumothorax-segmentation/data?select=download_images.py