[Day 42] Creating a UNet with PyTorch

 Hello :) 
Today is Day 42!


Quick summary of today:
  • Road segmentation with UNet
  • Forest area segmentation with UNet
  • Human segmentation with UNet


But why? I can easily load an UNet (well-made UNet, even pretrained) and it will work great on the images I found. Well, I tried to follow the model from yesterday, that was written with TensorFlow, and I just wanted to translate it to PyTorch and use it on new data. 

Was my attempt successful? - Not exactly haha 
Why ? - Training takes a lot of time. I ran out of free GPU resources and started using CPU (super slow ㅜㅜ). Adjusting the learning rate helped, but adjusting other hyper params can help too, but training time is too long without free GPU hours ㅜㅜ

I had 3 attempts on 3 different datasets
This is the format that I tried to follow.

Model structure:
Actually, I followed a TF model and translated it, so a potential issue is that I might have missed or did something in the wrong order. But given that the results are not terrible, such possibility is low (but still exists). 

For the loss, I used DiceLoss (also learned about it yesterday) and also binary crossentropy.
As for data augmentations - used albumentations and did horizontal and vertical flip with 50% probability. And of course resizing to 128 (or 256).


Below are summaries and info about the 3 models.

1) Road segmentation with UNet
Actually this dataset was a bit problematic, because some of the pictures had these white spots, which ruined the models' performance, but if we look in the ground truth, there are actually roads there.
Some results that were not too bad ~ around 0.5 loss. 


And here is an example of a picture that is not very good and just hurts the model (probably can just delete those from the training to improve performance)


2) Forest area segmentation with UNet
Example of this particular dataset ^
The valid loss after 25 epochs was 0.65 and a sample result. Definitely not great. The model needs more adjustment for these images. 

3) Human segmentation with UNet

Sample image (flipped in this case haha)
After 100 epochs, with learning rate 0.0001 (last attempt) - 0.6 loss
And a sample result
Not great. :(


Because the training was taking long, I ran notebooks at the same time on Kaggle, google colab and my laptop. Tried adjusting learning rate, image size, batch size. I tried to avoid adjusting the filters layers of the UNet to see if the same structure can work for different images, but unfortunately today I could not get good performance. I am thinking of concentrating on 1 and just let it run in the background while learning/trying something else tomorrow.



That is all for today!
See you tomorrow :) 

Popular posts from this blog

[Day 198] Transactions Data Streaming Pipeline Porject [v1 completed]

Image
 Hello :) Today is Day 198! A quick summary of today: data streaming pipeline project [v1 done] Here is a link to the project's repo. Well ... I did not know I can do it in a day (~14 hours) after yesterday's issues but here we are. Turns out in order to insert the full (~70 variables with nested/list structure), I need the proper pyspark schema. And yesterday I did not have that and that is why when I was reading data in the kafka producer I was getting NULL in the columns - my schema was wrong. Well today I not only fixed the schema for the 4 variables I had yesterday, but included *all* the variables that come from the Stripe API ~ 70 (for completeness).  When I run docker-compose, the data streams and is input into the postgres db (and is still running). Unfortunately, the free Stripe API for creating realistic transactions has a limit of 25, so every 3 seconds, 25 new transactions are sent to the db. It has been running half the day (since I got that set up) and as I am wr
Read more

[Day 107] Transforming natural language to charts

Image
 Hello :) Today is Day 107! A quick summary of today: created text2chart - transforming natural language to charts [ github ] [ webapp ] I am super sick, but in the around 3 hours of my fever not being high, I managed to study a bit. I was again thinking about creating some nlp project but was stuck on the idea of how to host a model. After looking around I found this article where they create a similar app called chat2vis  and they use codellama. I know codellama is an open source model that I can get from huggingface and I wondered, how did they host it. Turns out - they ask the user to input a huggingface api key, which then allows for querying and generating content. (AWESOME! this is applicable to some of the other PDF chat apps that I have, but I will look to make an app of them later).  My fever eased a bit, and I sat on my chair to do some work ^^ Here is a breakdown of the app. I used streamlit. Also, 2 default datasets during dev: Financial Statements of Major Companies(2009-
Read more

[Day 54] I became a backprop ninja! (woohoo)

Image
 Hello :) Today is Day 54! A quick summary of today: Read through cs231n's neural network lectures [1] [2] [3] Spent 8 hours going over Andrej Karpathy's backprop ninja tutorial After yesterday's confidence boost in doing backprop, I came in today with the intention of becoming of, as Andrej Karpathy calls it, a backprop ninja.  The cs231n NN lectures were the usual about how NN became to be, param initialization, activation functions, loss functions, regularization and optimizers. Nothing out of the ordinary, but still was good to reassure myself of that knowledge. But that was definitely not the main even of today. As I mentioned, I wanted to become (or at least attempt to become) a backprop ninja today. I have watched that particular tutorial twice, the 1st time I had no idea what is going on, the 2nd time I could follow, and now it was the 3rd time - and now, I wanted to participate and actually see the why and the how of the things we have to do, when we do manual back
Read more