Funny_muscle_enhancer :)
1.Discription:
- This is just a funny project that we want to see AutoEncoder (AE) can actually work on the some features. We will start to improve...
- Origial image and results for iterating 1~5-times.
2.Future work:
- Add more data to train
- Higher resolution
- Imporve preprocess (Inverse_Muscle_filter.ipynb): denoise process/enhance not incude face
- postprocess: brightness fixed/denoise process.
3.Usage:
- Downloads pre-trained model and put in the folder.
- Open "Pred.ipynb" .
- Input the image name you wwant to test.
- Run the whole code.
4.Training by yourself
- Downloads a lot of muscle image from internet (Since the copy right problem, I cannot share my dataset with you). The images type can be jpg/png/jfif/... . Notice: The more visible the muscle lines are in the images, the better. In our case, we have 204 images now.
- Download repository-skin detector-1 and skin detector-2. Then, put their with code.
- Create 2 folders: before_filtering/after_filtering. Put the downloaded images in to "after_filtering" folder. Also create 2 empty folders: before_filtering_rm_bg/after_filtering_rm_bg which will load images from Inverse_Muscle_filter.ipynb
- Run Inverse_Muscle_filter.ipynb.
- Open training.ipynb and run the code with suitable epochs.
5. Update History:
- [2021/12/21]
- Add 100+ image into dataset and remove gray style image (skin detector will not work).
- Modify preprocess to can output double images (weakening/original version with/without background). Original code just can output weakening/original version with background.
- Change training/prediction shape from (224,224,3) to (448,448,3).
- Training a model with size:(448,448,3) and put the new model in pre-trained model
- Training condition as shown as following: (x: epoch num, y: mse error of pixels)