Pytorch สำหรับการแบ่งส่วนความหมาย
ที่เก็บนี้มีบางรุ่นสำหรับการแบ่งส่วนความหมายและไปป์ไลน์ของรูปแบบการฝึกอบรมและการทดสอบที่ใช้ใน Pytorch
แบบจำลอง
- Vanilla FCN: FCN32, FCN16, FCN8 ในเวอร์ชันของ VGG, Resnet และ Densenet ตามลำดับ (เครือข่าย convolutional อย่างเต็มที่สำหรับการแบ่งส่วนความหมาย)
- U-NET (U-NET: เครือข่าย convolutional สำหรับการแบ่งส่วนภาพชีวการแพทย์)
- SEGNET (SEGNET: สถาปัตยกรรม DECODER DECODER DECODER สำหรับการแบ่งส่วนภาพ)
- PSPNET (เครือข่ายการแยกวิเคราะห์ฉากพีระมิด)
- GCN (เรื่องเคอร์เนลขนาดใหญ่)
- DUC, HDC (ทำความเข้าใจกับการแบ่งแยกความหมาย)
ความต้องการ
- Pytorch 0.2.0
- Tensorboard สำหรับ Pytorch ที่นี่เพื่อติดตั้ง
- ห้องสมุดอื่น ๆ (ค้นหาสิ่งที่คุณพลาดเมื่อเรียกใช้รหัส: -p)
การตระเตรียม
- ไปที่ไดเรกทอรี โมเดล และตั้งค่าเส้นทางของโมเดลที่ผ่านการฝึกอบรมใน config.py
- ไปที่ไดเรกทอรี ชุดข้อมูล และทำตาม readme
สิ่งที่ต้องทำ
- Deeplab v3
- โรงกลั่น
- ชุดข้อมูลเพิ่มเติม (เช่น ADE)