| | |
|---|
| | | if len(res_a) > 0: |
|---|
| | | sorted_results = sorted(res_a, key=itemgetter("id"), reverse=True) |
|---|
| | | # 查询前N个最大的ID |
|---|
| | | res_a = sorted_results[:int(self.config.get("detectnum"))-1] |
|---|
| | | res_a = sorted_results[:int(self.config.get("detectnum"))] |
|---|
| | | res_data = [] |
|---|
| | | for res in res_a: |
|---|
| | | data = { |
|---|
| | |
|---|
| | | image_id = self.collection.upsert(data).primary_keys |
|---|
| | | res['id'] = image_id[0] |
|---|
| | | res_data.append(res) |
|---|
| | | # logging.info(f"读取图像成功: {image_id}") |
|---|
| | | self.pool.submit(res_data) |
|---|
| | | # image_id = pool.tark_do(image_id,self.config.get("ragurl"),self.config.get("ragmode"),self.config.get("max_tokens")) |
|---|
| | | # logging.info(f"处理图像成功: {image_id}") |
|---|
