8
8
use function Codewithkyrian \Transformers \Utils \{memoryPeak , memoryUsage , timeUsage };
9
9
10
10
require_once './bootstrap.php ' ;
11
- $ modelConfig = ['model_type ' => 'vit ' ];
12
- $ processorConfig = [
13
- 'do_normalize ' => true ,
14
- 'do_pad ' => false ,
15
- 'do_rescale ' => true ,
16
- 'do_resize ' => true ,
17
- 'image_mean ' => [0.5 , 0.5 , 0.5 ],
18
- 'feature_extractor_type ' => "ImageFeatureExtractor " ,
19
- 'image_std ' => [1 , 1 , 1 ],
20
- 'resample ' => 2 ,
21
- 'rescale_factor ' => 0.00392156862745098 ,
22
- 'size ' => ['width ' => 1024 , 'height ' => 1024 ],
23
- ];
24
-
25
- $ model = AutoModel::fromPretrained (modelNameOrPath: 'briaai/RMBG-1.4 ' , config: $ modelConfig );
11
+
12
+ $ model = AutoModel::fromPretrained (modelNameOrPath: 'briaai/RMBG-1.4 ' );
26
13
$ processor = AutoProcessor::fromPretrained (modelNameOrPath: 'briaai/RMBG-1.4 ' );
27
14
28
- $ url = __DIR__ . '/../images/multitask.png ' ;
15
+ $ url = __DIR__ . '/../images/woman-w-bag.jpeg ' ;
29
16
30
17
$ image = Image::read ($ url );
31
18
34
21
['pixel_values ' => $ pixelValues ] = $ processor ($ image );
35
22
36
23
['output ' => $ output ] = $ model (['input ' => $ pixelValues ]);
37
-
24
+ //
38
25
$ mask = Image::fromTensor ($ output [0 ]->multiply (255 ))->resize ($ image ->width (), $ image ->height ());
39
-
26
+ //
40
27
$ mask ->save ($ fileName . '-mask.png ' );
41
-
28
+ //
42
29
$ maskedImage = $ image ->applyMask ($ mask );
43
-
44
- $ maskedImage ->save ($ fileName . '-masked.png ' );
30
+ //
31
+ // $maskedImage->save($fileName . '-masked.png');
0 commit comments