使用Flask搭建的后台
主要逻辑是先用yolo框选出物品,再用网络进行相似搜索
yolo是基于某大佬的项目:keras-yolo3
尝试过的网络有ProtoNet,Densenet, TripleLossNet
- 2019/12/14 部分绝对路径改成相对路径,添加yolo的模型文件的网盘地址
- Python 3
- PyTorch
- torchvision
- Tensorflow
- numpy
- Pillow
- opencv
- Flask
选择合适的网络的入口后直接运行,开启Flask后使用Postman测试接口,接口集中在入口文件中,app-PTNET.py
中的最为完整