search

LAION-AI / CLIP-based-NSFW-Detector

Other
299 stars 28 forks source link

Cleanup the codebase to make it easier to understand and use #11

Open HCA97 opened 1 year ago

HCA97 commented 1 year ago

Hi,

Thank you for your amazing work.

Recently, I was working on NSFW detection and came across your repository. I found it challenging to use AutoKeras models due to the outdated packages required to run them, see issue-9. As a result, I decided to clean up the repository and transition to PyTorch, making it easier for others to utilize these models more effectively. Furthermore, I have added additional resources and documentation to enhance understanding and usability.

As an outsider, I am unsure of the meaning of all the files, such as safety_setting.yaml therefore I keep some of the old files in the .old folder. Additionally, some of the information I provided might be incorrect, for example, details about the training dataset in the readme.

Best Regards HCA

rom1504 commented 1 year ago

clip retrieval https://github.com/rom1504/clip-retrieval is depending on the model files being where they were so please don't merge as is

HCA97 commented 1 year ago

Hi,

Thank you for your quick response and for pointing out the model files. I have now moved the model files back to their original location.

By the way, I don't want to cause any extra work for you. I know you guys have a lot going on. If you believe the changes to the repository are not needed and are unnecessary, please feel free to close the pull request. However, if you think the changes can be useful or find them necessary, and if there are still changes that need to be made, please let me know. I am happy to help.

Thanks