suryaprakash01 commited on
Commit
6529463
·
verified ·
1 Parent(s): 2e84597

Update readme.txt

Browse files
Files changed (1) hide show
  1. readme.txt +32 -0
readme.txt CHANGED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 3D Object Dimension & Volume Estimator
2
+
3
+ This project is a Streamlit web application that estimates the 3D dimensions (Length, Width, Height) and Volume of objects from user-uploaded images. It leverages Detectron2 for object detection and instance segmentation, and a custom Convolutional Neural Network (CNN) trained on the Pix3D dataset for dimension regression.
4
+
5
+ ## Features
6
+
7
+ * Upload single or multiple images of an object (different views).
8
+ * Detects objects using a pre-trained Detectron2 (Mask R-CNN) model.
9
+ * Displays segmentation masks and 2D bounding boxes for detected objects.
10
+ * For the largest detected object in each view:
11
+ * Crops the object using its segmentation mask.
12
+ * Feeds the cropped patch to a custom CNN to predict dimensions (L, W, H, V).
13
+ * Displays individual dimension predictions for each view.
14
+ * Calculates and displays aggregated (averaged) dimensions if multiple views are provided.
15
+ * User-friendly web interface built with Streamlit.
16
+
17
+ ## Models Used
18
+
19
+ 1. **Object Detection & Segmentation:**
20
+ * **Detectron2 (Mask R-CNN R50-FPN 3x):** Pre-trained on the COCO dataset. Used to identify objects and generate pixel-wise segmentation masks.
21
+ 2. **Dimension Estimation:**
22
+ * **Custom CNN (ResNet50 backbone):** Trained on image patches derived from the Pix3D dataset. The model takes a cropped image patch of an object as input and outputs its estimated Length, Width, Height (in meters), and Volume (in meters³).
23
+
24
+ ## Setup and Installation
25
+
26
+ Follow these steps to set up and run the application locally:
27
+
28
+ **1. Clone this GitHub Repository:**
29
+
30
+ ```bash
31
+ git clone [https://github.com/YourUsername/YourGitHubRepoName.git](https://github.com/YourUsername/YourGitHubRepoName.git)
32
+ cd YourGitHubRepoName