This is very nicely organized and we love the visual you included with how each piece of code is related to the project. The installation section is easy to follow. Your figures are well labeled and the code itself is nice with good comments throughout. The main recommendation we have would be to elaborate on the methodology more in your Readme (which models did you test) and link the figures used. While you mention using dvc for the model images, it would be good to link to the figures folder that you have within the code folder as well for a faster reference. Also, don't forget to include a summary of findings in the ReadMe as well. We're excited to see how this project turns out!
Comment from TAs and myself:
This is very nicely organized and we love the visual you included with how each piece of code is related to the project. The installation section is easy to follow. Your figures are well labeled and the code itself is nice with good comments throughout. The main recommendation we have would be to elaborate on the methodology more in your Readme (which models did you test) and link the figures used. While you mention using dvc for the model images, it would be good to link to the figures folder that you have within the code folder as well for a faster reference. Also, don't forget to include a summary of findings in the ReadMe as well. We're excited to see how this project turns out!