hab-spc / hab-ml

Harmful algae bloom CNN detection model, developed in PyTorch
6 stars 1 forks source link

Technical Debts #45

Open ktl014 opened 4 years ago

ktl014 commented 4 years ago

List below things we need to refactor for the hab-ml codebase

ktl014 commented 4 years ago
  1. Standard logger for logging error/debug msgs
  2. Get_image_dir un_hardcoding during model deployment
  3. Joining csv files and jsons together after deployment
  4. giving read/write access for all main directories
  5. argument parser vs config file
ktl014 commented 4 years ago
  1. probabilities in csv files should only take top 5 probabilities
zhouyuanyuan1999 commented 4 years ago
  1. Add features to be able to query the right model_dir easier. Improve model_dir naming convention.
ktl014 commented 4 years ago
  1. add unique id to predictions.json file in deploy mode
ktl014 commented 4 years ago

Update for number 5:

@igneel73 any comments on this?

ktl014 commented 4 years ago

![Uploading image.png…]()

ktl014 commented 4 years ago

Remove spc submodule from hab_ml

Eighteensi commented 2 months ago

I would like to replicate this model, but I do not have the image recognition data and labels. Do you still have the original data from back then? I would like to try out the model's performance for further research. Thank you