/Deecamp-SimilarGoodsSearcher

Deecamp项目小程序后端。使用Flask作为Web框架，yolo进行框选，使用Protonet等网络进行分类，

Primary LanguageJupyter Notebook

Deecamp 2019 项目实践后端部分

使用Flask搭建的后台

主要逻辑是先用yolo框选出物品，再用网络进行相似搜索

yolo是基于某大佬的项目：keras-yolo3

尝试过的网络有ProtoNet，Densenet， TripleLossNet

  • 2019/12/14 部分绝对路径改成相对路径，添加yolo的模型文件的网盘地址

Requirements

  • Python 3
  • PyTorch
  • torchvision
  • Tensorflow
  • numpy
  • Pillow
  • opencv
  • Flask

Usage

选择合适的网络的入口后直接运行，开启Flask后使用Postman测试接口，接口集中在入口文件中，app-PTNET.py 中的最为完整