This project is a scene classification model using Resnet50 based on the PaddlePaddle framework. It has been trained using the Place365 dataset, and it achieved a Top1 accuracy of 53% and a Top10 accuracy of 81% on this dataset.
Here are the steps to install the project:
-
Clone the project repository:
git clone https://github.com/shaohon/place365classification.git
-
Enter the project directory:
cd place365classification
-
Install the requirements:
pip install -r requirements.txt
Follow the next steps to run locally.
-
Download weights in download link and put the file in the project root.
-
Run the prediction script in local:
python app.py
You can use SwanAPI to package the model as an API service.
First, install swanapi using the following command
pip install swanapi -i https://pypi.org/simple
Then, you can directly turn the model into a prediction service.
python predict.py
You can also build a deep learning inference image with just one command.
swanapi build -t my-dl-model
Run python tools/post.py
to test whether the service is online
Your contributions to improve this project are always welcome. You can participate by following these steps:
- Fork this repo to your own GitHub account.
- Make a new branch and make your changes:
git checkout -b your_branch_name
- Commit your changes:
git commit -m 'Add some features'
- Push your branch to your GitHub repo:
git push origin your_branch_name
- Submit a Pull Request and wait for review.
If you ever encounter any issues while using the project, or have any suggestions, please don't hesitate to contact us:
- Author: shaohon
- Email: shaohon.chen@swanhub.co
Thank you for your support!
This project is released under the GNU General Public License (GPL). This is a widely used free software license, which guarantees end users the freedom to run, study, share, and modify the software.
For the full license text, please see the LICENSE file in the root directory of this source tree. If the LICENSE file does not exist, you can add one. Note that you should include the full GPL license text in this LICENSE file.
This ends the README of the scene classification open source project, and I hope you find it helpful. If you need additional information or have any other needs, please let us know.