ดาวน์โหลด pytorch_classification - pytorch_classification ซอร์สโค้ดดาวน์โหลดดาวน์โหลด

pytorch_classification

หลาม

1.0.0

ดาวน์โหลด

การแนะนำ

ฟังก์ชั่นการจำแนกรูปภาพ Pytorch ที่ใช้ Torchision นำมาใช้

อัปเดตล่าสุด

อัปเดต 2022.11.05
- โซลูชันการอนุมาน Tensorrt C ++ ที่เพิ่มขึ้นใหม่
อัปเดตในปี 2022.10.29 การปรับโครงสร้างรหัสฟังก์ชั่นพื้นฐานนั้นเหมือนกัน
- สนับสนุนการฝึกอบรม Pytorch DDP
- การให้เหตุผลแบบจำลองที่รองรับ C ++ Libtorch
- สนับสนุนสคริปต์ที่ทำงานในคลิกเดียว
- เพิ่มโมดูลบันทึก

หากคุณคุ้นเคยกับเวอร์ชันก่อนหน้าโปรดดูรหัสของเวอร์ชัน V1: V1 เวอร์ชัน

ฟังก์ชั่นหลัก:

การใช้ pytorch เพื่อตระหนักถึงการจำแนกประเภทภาพโดยใช้ Torchision สามารถขยายเครือข่ายการจำแนกภาพเช่น DensityNet, Resnext, Mobilenet, EfficientNet, Swin Transformer ฯลฯ

หากมีประโยชน์ยินดีต้อนรับสู่สตาร์

ใช้ฟังก์ชั่น

ฟังก์ชั่นพื้นฐานใช้ pytorch เพื่อตระหนักถึงการจำแนกรูปภาพ
รวมถึงการปรับอัตราการเรียนรู้แบบโคไซน์ด้วยการอุ่นเครื่อง
อัตราการเรียนรู้ขั้นตอนของการอุ่นเครื่องได้รับการปรับให้เหมาะสม
การทำนายฟิวชั่นแบบหลายรูปแบบการถ่วงน้ำหนักและการลงคะแนนฟิวชั่น
ใช้ Flask + Redis เพื่อใช้การปรับใช้ Model Cloud API (Tag V1)
การปรับใช้แบบจำลองของ C ++ libtorch
การเพิ่มการทำนายเมื่อใช้การทดสอบ TTA (TAG V1)
การเพิ่มการใช้งาน pytorch ของฉลากเรียบ (แท็กเรียบ) (แท็ก v1)
เพิ่มไปยังคุณสมบัติแยกโดยใช้ CNN และจำแนกโดยใช้ SVM, RF, MLP, KNN และตัวแยกประเภทอื่น ๆ (TAG V1)
แสดงภาพเลเยอร์คุณสมบัติ

สภาพแวดล้อมที่ใช้งาน

Python3.7
Pytorch 1.8.1
Torchvision 0.9.1
OpenCV (ใช้การอนุมาน Libtorch CPP, เวอร์ชัน 3.4.6) (ไม่บังคับ)
Libtorch CPP ใช้เหตุผล (ไม่บังคับ)

เริ่มต้น

แบบฟอร์มชุดข้อมูล

รูปแบบการจัดระเบียบของชุดข้อมูลอ้างถึง sample_files/imgs/listfile.txt

ทดสอบการฝึกอบรม

แก้ไขพารามิเตอร์ใน run.sh และ run run.sh โดยตรง

พารามิเตอร์ที่แก้ไขแล้วหลัก:

 OUTPUT_PATH 模型保存和log文件的路径

TRAIN_LIST 训练数据集的list文件
VAL_LIST  测试集合的list文件
model_name 默认是resnet50
lr 学习率
epochs 训练总的epoch
batch-size  batch的大小
j dataloader的num_workers的大小
num_classes 类别数

การอนุมาน Libtorch

รหัสถูกเก็บไว้ในโฟลเดอร์ cpp_inference

ใช้ CPP_Inference/traced_model/trace_model.py เพื่อส่งออกโมเดลที่ผ่านการฝึกอบรม
รวบรวมรหัส openCV และ libtorch ที่ต้องการเป็น cpp_inference/third_party_library
การรวบรวม