create_training_pipeline - Githubissues

hab-spc / hab-ml

Harmful algae bloom CNN detection model, developed in PyTorch

6 stars 1 forks source link

create_training_pipeline #23

Closed ktl014 closed 5 years ago

ktl014 commented 5 years ago

Task description: Given a set time window, we should be able to download images using spici and assign the labels from the 'user_labels' and 'tags' to create our dataset for our model. The csv files from the SPICI should be configured to be input into 'SPCHABDataset'.

DoD:

[x] write script for this functionality
[x] script should be able to specify which time window to pull from
[x] script should also be able to identify which labels are acceptable to assign each image
[x] script should output csv in correct format for 'SPCHABDataset'

ktl014 commented 5 years ago

Items 1 and 4 done. 2 and 3 are also done, but we should refactor it a little bit for better design and implementation.