在Flask应用中处理无关的骨折检测上传

Question

我正在开发一个Flask应用程序，用于检测骨折。用户可以上传X光图像，应用程序会预测是否存在骨折。不过，我遇到了一个问题，就是如何处理那些不相关的上传，比如上传肘部、手部和肩部以外的身体部位的图片。

下面是我Flask应用程序代码的概述：

Flask应用程序代码

import os
from flask import Flask, request, jsonify
from werkzeug.utils import secure_filename
from predictions import predict
from flask_cors import CORS

app = Flask(__name__)
CORS(app)  # Enable CORS for all routes
app.config['UPLOAD_FOLDER'] = 'uploads'
app.config['ALLOWED_EXTENSIONS'] = {'png', 'jpg', 'jpeg'}

def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in app.config['ALLOWED_EXTENSIONS']

@app.route('/predict', methods=['POST'])
def predict_bone_fracture():
    if 'file' not in request.files:
        return jsonify({'error': 'No file part'})

    file = request.files['file']

    if file.filename == '':
        return jsonify({'error': 'No selected file'})

    if file and allowed_file(file.filename):
        filename = secure_filename(file.filename)
        filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
        file.save(filepath)

        # Perform prediction using the predict function from predictions.py
        bone_type_result = predict(filepath)
        result = predict(filepath, bone_type_result)

        # You can customize the response based on your requirements
        return jsonify({'bone_type': bone_type_result, 'result': result})

    return jsonify({'error': 'Invalid file format'})

if __name__ == "__main__":
    app.run()

predict_bone_fracture()这个函数接收用户上传的图片，把它保存到指定的文件夹，然后使用一个外部模块(predictions.py)里的predict()函数进行预测。如果上传的文件不是图片或者格式不支持，它会返回一个错误信息。

我最担心的是如何处理用户上传的那些与指定身体部位不符的图片（也就是肘部、手部、肩部）。比如，如果用户上传了一张眼睛的图片，而不是骨头的图片，应用程序应该拒绝这个上传，并给出一个合适的错误提示。

我觉得我需要加入一个机制，来检测上传图片中的相关身体部位，并确认它们是否符合预测所需的身体部位（也就是肘部、手部、肩部）。不过，我不太确定该用什么方法来实现这个功能。

你能给我一些建议或者想法，来解决这个问题吗？具体来说，我希望得到以下方面的指导：

如何实现一个机制，来检测上传图片中的相关身体部位。

如何检查检测到的身体部位是否与预测所需的身体部位相符。

如何为不相关的上传提供合适的错误处理和提示信息。

任何见解或代码示例都将非常感谢。谢谢！