Export onnx file Use script directly import torch from mmdet.apis import init_detector, inference_detector config_file = ‘./configs/yolo/yolov3_mobilenetv2_8xb24-ms-416-300e_coco.py’ checkpoint_file = ‘yolov3_mobilenetv2_mstrain-416_300e_coco_20210718_010823-f68a07b3.pth’ model = init_detector(config_file, checkpoint_file, device=’cpu’) # or device=’cuda:0′ torch.onnx.export(model, (torch.zeros(1, 3, 416, 416),), “yolov3.onnx”, opset_version=11) The exported onnx structure is as follows: The output is the output of three different levels of detection heads. If you […]
Tag: mode
Digital modeling experience-data processing-pandas
Digital analog experience-data processing-pandas Detailed explanation of the code: will be added next time import pandas as pd import numpy as np # # Set panda display function # pd.set_option(‘display.max_columns’, 10) # pd.set_option(‘display.max_rows’, 100) # pd.set_option(‘display.width’, 100) Series basic operations obj=pd.Series([4,7,-5,3]) obj 0 4 1 7 2-5 3 3 dtype: int64 obj.values array([ 4, 7, […]
Generative AI – Knowledge Graph Prompting: A multi-document question answering method based on large models
Large language models (LLMs) have revolutionized natural language processing (NLP) tasks. They change the way we interact with and process text data. These powerful AI models, such as OpenAI’s GPT-4, have changed the way human-like text is understood and generated, leading to numerous breakthrough applications across a variety of industries. LangChain is an open source […]
Teach you step by step how to implement the producer-consumer model
?Yin’s Blog | Yin’s Blog (yinkai.cc), welcome to my blog to read. This article will introduce you to the definition, characteristics, and processes of the producer-consumer model, and take you step by step to implement the producer-consumer model. 1. Introduction The producer-consumer pattern is a concurrency design pattern used to solve the collaboration and data […]
CNN model training–Code practice for verification code identification
Tip: After the article is written, the table of contents can be automatically generated. For how to generate it, please refer to the help document on the right. Article directory Summary Abstract CNN model training–Verification code identification 1. Code practice 2. Clarify training needs 3. Collect training data 4. CNN network architecture 5. Model training […]
PureEdgeSim -locationmanager package -MobilityModel file
MobilityModel The main class of the mobility manager module, which generates mobility paths for different edge devices. Current device location protected Location currentLocation; Is the device moving? protected boolean isMobile = false; Minimum pause duration protected double minPauseDuration; Maximum pause duration protected double maxPauseDuration; Maximum movement duration protected double maxMobilityDuration; Minimum movement duration protected double […]
Hive data warehouse environment construction and data warehouse modeling
Summary of the problem: After I completed the data warehouse modeling, one day I opened my dbeaver again and all the data was gone. I found many solutions but could not solve my problem. In a fit of anger, I deleted hive and all tables. Start over 1. Hive data warehouse environment construction Hive installation […]
[Large model] Large language model corpus download
Article directory Overview Hugging Face obs operation git-lfs example RedPajama-Data-1T SlimPajama-627B/ git clone resume Data Format References Overview Corpus is very important in large model training. Currently, there are various corpora available for download on the public Internet, but it is impossible for every user and every training task. To pull corpus through the public […]
Modeling analysis of the “Friends” sitcom data set based on LDA topic analysis
?♂? Personal homepage: @ aiperson’s personal homepage ?About the author: Python learner I hope everyone will support us and we will make progress together! If the article is helpful to you, Welcome to comment Like Collection Add follow + Table of Contents 1.Project background 2. Introduction to data sets 3.Technical Tools 4. Experimental process 4.1 […]
How to use Trend Cloud GPU to run AI project online – deploy the latest ChatGLM3-6B model
Learning tutorial 1. Get free computing power Datawhale exclusive registration: Trend Cloudhttps://growthdata.virtaicloud.com/t/vs 2. Deploy the latest ChatGLM3-6B model 1. Create project After creating an account, enter your own space and click Create Project in the upper right corner. Give the project a name you like and choose local code Mirror selection pytorch2.0.1, python3.9 Select the […]